Home
last modified time | relevance | path

Searched refs:OH_NN_STACK_AXIS (Results 1 – 9 of 9) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dstack_builder_test.cpp76 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STACK_AXIS);
94 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STACK_AXIS);
113 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STACK_AXIS);
131 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STACK_AXIS);
164 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STACK_AXIS);
182 SaveAxisTensor(OH_NN_INT32, paramDim, nullptr, OH_NN_STACK_AXIS);
200 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STACK_AXIS);
219 …shared_ptr<NNTensor> axisTensor =TransToNNTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STACK_AXIS);
239 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STACK_AXIS);
287 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STACK_AXIS);
/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/
H A Dstack_builder.h44 {OH_NN_STACK_AXIS, &StackBuilder::SetAxis}
/ohos5.0/foundation/ai/neural_network_runtime/interfaces/kits/c/neural_network_runtime/
H A Dneural_network_runtime_type.h2972 OH_NN_STACK_AXIS = 53, enumerator
/ohos5.0/docs/zh-cn/application-dev/reference/apis-neural-network-runtime-kit/
H A Dneural__network__runtime__type_8h.md71 …UM = 50, OH_NN_SPLIT_SIZE_SPLITS = 51,<br/>OH_NN_SQUEEZE_AXIS = 52, OH_NN_STACK_AXIS = 53, OH_NN_S…
H A D_neural_network_runtime.md72 …UM = 50, OH_NN_SPLIT_SIZE_SPLITS = 51,<br/>OH_NN_SQUEEZE_AXIS = 52, OH_NN_STACK_AXIS = 53, OH_NN_S…
827 | OH_NN_STACK_AXIS | 当张量作为Stack算子的Axis参数时，使用本枚举值。 |
H A D_neural_nework_runtime.md72 …UM = 50, OH_NN_SPLIT_SIZE_SPLITS = 51,<br/>OH_NN_SQUEEZE_AXIS = 52, OH_NN_STACK_AXIS = 53, OH_NN_S…
827 | OH_NN_STACK_AXIS | 当张量作为Stack算子的Axis参数时，使用本枚举值。 |
/ohos5.0/docs/en/application-dev/reference/apis-neural-network-runtime-kit/
H A Dneural__network__runtime__type_8h.md71 …NUM = 50, OH_NN_SPLIT_SIZE_SPLITS = 51,<br>OH_NN_SQUEEZE_AXIS = 52, OH_NN_STACK_AXIS = 53, OH_NN_S…
H A D_neural_network_runtime.md72 …NUM = 50, OH_NN_SPLIT_SIZE_SPLITS = 51,<br>OH_NN_SQUEEZE_AXIS = 52, OH_NN_STACK_AXIS = 53, OH_NN_S…
827 | OH_NN_STACK_AXIS | Used when the tensor is used as the **Axis** parameter of the Stack operator. …
/ohos5.0/foundation/ai/neural_network_runtime/example/deep_learning_framework/tflite/delegates/nnrt_delegate/
H A Dnnrt_op_builder.cpp210 mappingArgs.builder->AddScalarInt64Tensor(axis, OH_NN_STACK_AXIS); in AddPackParams()