Searched refs:OH_NN_OPS_ARG_MAX (Results 1 – 7 of 7) sorted by relevance
/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/ |
H A D | argmax_builder.cpp | 169 REGISTER_OPS(ArgMaxBuilder, OH_NN_OPS_ARG_MAX);
|
/ohos5.0/foundation/ai/neural_network_runtime/interfaces/kits/c/neural_network_runtime/ |
H A D | neural_network_runtime_type.h | 1683 OH_NN_OPS_ARG_MAX = 54, enumerator
|
/ohos5.0/docs/zh-cn/application-dev/reference/apis-neural-network-runtime-kit/ |
H A D | neural__network__runtime__type_8h.md | 70 …LL = 51, OH_NN_OPS_QUANT_DTYPE_CAST = 52,<br/>OH_NN_OPS_TOP_K = 53, OH_NN_OPS_ARG_MAX = 54, OH_NN_…
|
H A D | _neural_network_runtime.md | 71 …LL = 51, OH_NN_OPS_QUANT_DTYPE_CAST = 52,<br/>OH_NN_OPS_TOP_K = 53, OH_NN_OPS_ARG_MAX = 54, OH_NN_… 629 | OH_NN_OPS_ARG_MAX | 返回跨轴的张量最大值的索引。<br/>输入:<br/>- input,n维张量,输入张量(N,∗),其中∗意味着任意数量的附加维度。<br/>参数:<br…
|
H A D | _neural_nework_runtime.md | 71 …LL = 51, OH_NN_OPS_QUANT_DTYPE_CAST = 52,<br/>OH_NN_OPS_TOP_K = 53, OH_NN_OPS_ARG_MAX = 54, OH_NN_… 629 | OH_NN_OPS_ARG_MAX | 返回跨轴的张量最大值的索引。<br/>输入:<br/>- input,n维张量,输入张量(N,∗),其中∗意味着任意数量的附加维度。<br/>参数:<br…
|
/ohos5.0/docs/en/application-dev/reference/apis-neural-network-runtime-kit/ |
H A D | neural__network__runtime__type_8h.md | 70 …ALL = 51, OH_NN_OPS_QUANT_DTYPE_CAST = 52,<br>OH_NN_OPS_TOP_K = 53, OH_NN_OPS_ARG_MAX = 54, OH_NN_…
|
H A D | _neural_network_runtime.md | 71 …ALL = 51, OH_NN_OPS_QUANT_DTYPE_CAST = 52,<br>OH_NN_OPS_TOP_K = 53, OH_NN_OPS_ARG_MAX = 54, OH_NN_… 629 | OH_NN_OPS_ARG_MAX | Returns the index of the maximum tensor value across axes.<br>Input:<br>- **i…
|