| /ohos5.0/docs/zh-cn/device-dev/reference/hdi-apis/nnrt/ |
| H A D | _arg_max_fusion_v10.md | 31 | long [topK](#topk) | 轴上前K个最大值。 |
69 ### topK subsection
72 long ArgMaxFusion::topK
|
| H A D | _arg_max_fusion_v20.md | 31 | long [topK](#topk) | 轴上前K个最大值。 |
69 ### topK subsection
72 long ArgMaxFusion::topK
|
| /ohos5.0/foundation/ai/ai_engine/services/server/plugin/cv/image_classification/source/ |
| H A D | ic_plugin.cpp | 50 int32_t GetTopK(const int32_t *data, size_t size, size_t topK, Items &result) in GetTopK() argument
56 if (topK > size) { in GetTopK()
57 topK = size; in GetTopK()
60 while (index < topK) { in GetTopK()
|
| /ohos5.0/drivers/interface/nnrt/v1_0/ |
| H A D | NodeAttrTypes.idl | 37 long topK;
|
| /ohos5.0/drivers/interface/nnrt/v2_0/ |
| H A D | NodeAttrTypes.idl | 37 long topK;
|
| /ohos5.0/foundation/ability/idl_tool/test/hdi_unittest/nnrt_v1_0/foo/nnrt/v1_0/ |
| H A D | NodeAttrTypes.idl | 37 long topK;
|
| /ohos5.0/foundation/ability/idl_tool/test/hdi_unittest/nnrt_v2_0/foo/nnrt/v2_0/ |
| H A D | NodeAttrTypes.idl | 37 long topK;
|
| /ohos5.0/docs/zh-cn/application-dev/reference/apis-neural-network-runtime-kit/ |
| H A D | _neural_network_runtime.md | 629 …维度。<br/>- keepDims，bool值，是否维持输入张量维度。<br/>- topK，要返回最大值的数量，默认为1，当topK等于1时，将返回输入张量中最大值的索引；当topK大于1时，…
898 | OH_NN_ARG_MAX_TOP_K<sup>12+</sup> | 当张量作为ArgMax算子的topK参数时，使用本枚举值。 |
|
| H A D | _neural_nework_runtime.md | 629 …维度。<br/>- keepDims，bool值，是否维持输入张量维度。<br/>- topK，要返回最大值的数量，默认为1，当topK等于1时，将返回输入张量中最大值的索引；当topK大于1时，…
898 | OH_NN_ARG_MAX_TOP_K<sup>12+</sup> | 当张量作为ArgMax算子的topK参数时，使用本枚举值。 |
|
| /ohos5.0/foundation/ability/idl_tool/test/hdi_unittest/nnrt_v1_0/dump_ast_target/ |
| H A D | dump.txt | 78 long topK;
|
| /ohos5.0/foundation/ability/idl_tool/test/hdi_unittest/nnrt_v2_0/dump_ast_target/ |
| H A D | dump.txt | 143 long topK;
|
| /ohos5.0/foundation/ability/idl_tool/test/hdi_unittest/nnrt_v1_0/cpp_target/nnrt/v1_0/ |
| H A D | node_attr_types.h.txt | 70 int64_t topK;
|
| /ohos5.0/foundation/ability/idl_tool/test/hdi_unittest/nnrt_v2_0/cpp_target/nnrt/v2_0/ |
| H A D | node_attr_types.h.txt | 70 int64_t topK;
|
| /ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ |
| H A D | lite_graph_to_hdi_model_v1_0.cpp | 81 argMaxFusion.topK = mindspore::lite::MindIR_ArgMaxFusion_GetTopK(primitive); in ConvertArgMaxFusion()
|
| H A D | lite_graph_to_hdi_model_v2_0.cpp | 81 argMaxFusion.topK = mindspore::lite::MindIR_ArgMaxFusion_GetTopK(primitive); in ConvertArgMaxFusion()
|
| H A D | lite_graph_to_hdi_model_v2_1.cpp | 98 argMaxFusion.topK = mindspore::lite::MindIR_ArgMaxFusion_GetTopK(primitive); in ConvertArgMaxFusion()
|
| /ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/device_manager/ |
| H A D | device_manager_test.cpp | 233 int64_t topK {1}; variable
236 …void* primitive = mindspore::lite::MindIR_ArgMaxFusion_CreatePrimitive(axis, topK, keepDims, outMa…
|
| /ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/inner_model/ |
| H A D | inner_model_test.cpp | 1151 int64_t topK {1}; variable
1154 …void* primitive = mindspore::lite::MindIR_ArgMaxFusion_CreatePrimitive(axis, topK, keepDims, outMa…
|
| /ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v2_0/inner_model/ |
| H A D | inner_model_test.cpp | 1090 int64_t topK {1}; variable
1093 …void* primitive = mindspore::lite::MindIR_ArgMaxFusion_CreatePrimitive(axis, topK, keepDims, outMa…
|
| /ohos5.0/docs/en/application-dev/reference/apis-neural-network-runtime-kit/ |
| H A D | _neural_network_runtime.md | 629 …topK**: number of maximum values to be returned. The default value is **1**. If **topK** is equal …
898 | OH_NN_ARG_MAX_TOP_K<sup>12+</sup> | Used when the tensor is used as the **topK** parameter of the…
|
| /ohos5.0/drivers/interface/nnrt/v2_1/ |
| H A D | NodeAttrTypes.idl | 177 long topK;
|