Home
last modified time | relevance | path

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

/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Donehot_builder_test.cpp69 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_ONE_HOT_AXIS);
84 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_ONE_HOT_AXIS);
104 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_ONE_HOT_AXIS);
122 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_ONE_HOT_AXIS);
162 nullptr, OH_NN_ONE_HOT_AXIS);
197 nullptr, OH_NN_ONE_HOT_AXIS);
213 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_ONE_HOT_AXIS);
H A Dpad_builder_test.cpp244 SetConstValueTensor(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_ONE_HOT_AXIS);
261 SetPaddingModeTensor(OH_NN_INT32, m_paramDim, nullptr, OH_NN_ONE_HOT_AXIS);
/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/
H A Donehot_builder.h42 {OH_NN_ONE_HOT_AXIS, &OnehotBuilder::SetAxis}
/ohos5.0/foundation/ai/neural_network_runtime/interfaces/kits/c/neural_network_runtime/
H A Dneural_network_runtime_type.h2937 OH_NN_ONE_HOT_AXIS = 42, enumerator
/ohos5.0/docs/zh-cn/application-dev/reference/apis-neural-network-runtime-kit/
H A Dneural__network__runtime__type_8h.md71 …_NN_MAX_POOL_ACTIVATION_TYPE = 40, OH_NN_MUL_ACTIVATION_TYPE = 41, OH_NN_ONE_HOT_AXIS = 42, OH_NN_…
H A D_neural_network_runtime.md72 …_NN_MAX_POOL_ACTIVATION_TYPE = 40, OH_NN_MUL_ACTIVATION_TYPE = 41, OH_NN_ONE_HOT_AXIS = 42, OH_NN_…
816 | OH_NN_ONE_HOT_AXIS | 当张量作为OneHot算子的axis参数时，使用本枚举值。 |
H A D_neural_nework_runtime.md72 …_NN_MAX_POOL_ACTIVATION_TYPE = 40, OH_NN_MUL_ACTIVATION_TYPE = 41, OH_NN_ONE_HOT_AXIS = 42, OH_NN_…
816 | OH_NN_ONE_HOT_AXIS | 当张量作为OneHot算子的axis参数时，使用本枚举值。 |
/ohos5.0/docs/en/application-dev/reference/apis-neural-network-runtime-kit/
H A Dneural__network__runtime__type_8h.md71 …_NN_MAX_POOL_ACTIVATION_TYPE = 40, OH_NN_MUL_ACTIVATION_TYPE = 41, OH_NN_ONE_HOT_AXIS = 42, OH_NN_…
H A D_neural_network_runtime.md72 …_NN_MAX_POOL_ACTIVATION_TYPE = 40, OH_NN_MUL_ACTIVATION_TYPE = 41, OH_NN_ONE_HOT_AXIS = 42, OH_NN_…
816 | OH_NN_ONE_HOT_AXIS | Used when the tensor is used as the **axis** parameter of the OneHot operato…