Home
last modified time | relevance | path

Searched refs:OH_NN_SCALE_AXIS (Results 1 – 17 of 17) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dscale_builder_test.cpp80 SaveAxisTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_SCALE_AXIS);
96 SaveAxisTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_SCALE_AXIS);
117 SaveAxisTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_SCALE_AXIS);
136 SaveAxisTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_SCALE_AXIS);
199 SaveAxisTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_SCALE_AXIS);
223 …hared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT64, axistDim, nullptr, OH_NN_SCALE_AXIS);
245 SaveAxisTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_SCALE_AXIS);
269 SaveAxisTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_SCALE_AXIS);
291 SaveAxisTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_SCALE_AXIS);
328 SaveAxisTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_SCALE_AXIS);
[all …]
H A Dtopk_builder_test.cpp268 SaveSortedTensor(OH_NN_BOOL, paramDim, nullptr, OH_NN_SCALE_AXIS);
289 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SCALE_AXIS);
H A Dsqueeze_builder_test.cpp221 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SCALE_AXIS);
H A Dstack_builder_test.cpp257 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SCALE_AXIS);
H A Dunsqueeze_builder_test.cpp256 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SCALE_AXIS);
H A Dsoftmax_builder_test.cpp258 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SCALE_AXIS);
H A Dsub_builder_test.cpp257 SaveActivateTypeTensor(OH_NN_INT8, paramDim, nullptr, OH_NN_SCALE_AXIS);
H A Dsplit_builder_test.cpp325 SaveSizeSplitsTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SCALE_AXIS);
H A Dspacetobatchnd_builder_test.cpp364 SavePaddingsTensor(OH_NN_INT64, paddingsDim, nullptr, OH_NN_SCALE_AXIS);
H A Dstrided_slice_builder_test.cpp508 SaveShrinkAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SCALE_AXIS);
/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/
H A Dscale_builder.h48 {OH_NN_SCALE_AXIS, &ScaleBuilder::SetAxis}
/ohos5.0/foundation/ai/neural_network_runtime/interfaces/kits/c/neural_network_runtime/
H A Dneural_network_runtime_type.h2947 OH_NN_SCALE_AXIS = 45, enumerator
/ohos5.0/docs/zh-cn/application-dev/reference/apis-neural-network-runtime-kit/
H A Dneural__network__runtime__type_8h.md71 …OH_NN_PAD_CONSTANT_VALUE = 43,<br/>OH_NN_SCALE_ACTIVATIONTYPE = 44, OH_NN_SCALE_AXIS = 45, OH_NN_S…
H A D_neural_network_runtime.md72 …OH_NN_PAD_CONSTANT_VALUE = 43,<br/>OH_NN_SCALE_ACTIVATIONTYPE = 44, OH_NN_SCALE_AXIS = 45, OH_NN_S…
819 | OH_NN_SCALE_AXIS | 当张量作为Scale算子的axis参数时，使用本枚举值。 |
H A D_neural_nework_runtime.md72 …OH_NN_PAD_CONSTANT_VALUE = 43,<br/>OH_NN_SCALE_ACTIVATIONTYPE = 44, OH_NN_SCALE_AXIS = 45, OH_NN_S…
819 | OH_NN_SCALE_AXIS | 当张量作为Scale算子的axis参数时，使用本枚举值。 |
/ohos5.0/docs/en/application-dev/reference/apis-neural-network-runtime-kit/
H A Dneural__network__runtime__type_8h.md71 … OH_NN_PAD_CONSTANT_VALUE = 43,<br>OH_NN_SCALE_ACTIVATIONTYPE = 44, OH_NN_SCALE_AXIS = 45, OH_NN_S…
H A D_neural_network_runtime.md72 … OH_NN_PAD_CONSTANT_VALUE = 43,<br>OH_NN_SCALE_ACTIVATIONTYPE = 44, OH_NN_SCALE_AXIS = 45, OH_NN_S…
819 | OH_NN_SCALE_AXIS | Used when the tensor is used as the **axis** parameter of the Scale operator. …