Searched refs:OH_NN_FLATTEN_AXIS (Results 1 – 8 of 8) sorted by relevance
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/ |
H A D | flatten_test.cpp | 69 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_FLATTEN_AXIS); 84 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_FLATTEN_AXIS); 104 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_FLATTEN_AXIS); 122 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_FLATTEN_AXIS); 163 nullptr, OH_NN_FLATTEN_AXIS); 199 nullptr, OH_NN_FLATTEN_AXIS); 215 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_FLATTEN_AXIS);
|
/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/ |
H A D | flatten_builder.h | 46 {OH_NN_FLATTEN_AXIS, &FlattenBuilder::SetAxis}
|
/ohos5.0/foundation/ai/neural_network_runtime/interfaces/kits/c/neural_network_runtime/ |
H A D | neural_network_runtime_type.h | 3065 OH_NN_FLATTEN_AXIS = 79, enumerator
|
/ohos5.0/docs/zh-cn/application-dev/reference/apis-neural-network-runtime-kit/ |
H A D | neural__network__runtime__type_8h.md | 71 …KEEPDIMS = 76, OH_NN_UNSQUEEZE_AXIS = 77, OH_NN_UNSTACK_AXIS = 78, OH_NN_FLATTEN_AXIS = 79,<br/>OH…
|
H A D | _neural_network_runtime.md | 72 …KEEPDIMS = 76, OH_NN_UNSQUEEZE_AXIS = 77, OH_NN_UNSTACK_AXIS = 78, OH_NN_FLATTEN_AXIS = 79,<br/>OH… 853 | OH_NN_FLATTEN_AXIS<sup>12+</sup> | 当张量作为Flatten算子的axis参数时,使用本枚举值。 |
|
H A D | _neural_nework_runtime.md | 72 …KEEPDIMS = 76, OH_NN_UNSQUEEZE_AXIS = 77, OH_NN_UNSTACK_AXIS = 78, OH_NN_FLATTEN_AXIS = 79,<br/>OH… 853 | OH_NN_FLATTEN_AXIS<sup>12+</sup> | 当张量作为Flatten算子的axis参数时,使用本枚举值。 |
|
/ohos5.0/docs/en/application-dev/reference/apis-neural-network-runtime-kit/ |
H A D | neural__network__runtime__type_8h.md | 71 …KEEPDIMS = 76, OH_NN_UNSQUEEZE_AXIS = 77, OH_NN_UNSTACK_AXIS = 78, OH_NN_FLATTEN_AXIS = 79,<br>OH_…
|
H A D | _neural_network_runtime.md | 72 …KEEPDIMS = 76, OH_NN_UNSQUEEZE_AXIS = 77, OH_NN_UNSTACK_AXIS = 78, OH_NN_FLATTEN_AXIS = 79,<br>OH_… 853 | OH_NN_FLATTEN_AXIS<sup>12+</sup> | Used when the tensor is used as the **axis** parameter of the …
|