Searched refs:outMaxValue (Results 1 - 7 of 7) sorted by relevance
| /foundation/ai/neural_network_runtime/test/unittest/ops/ |
| H A D | argmax_test.cpp | 317 int64_t* outMaxValue = new (std::nothrow) int64_t(0); in HWTEST_F() local
318 EXPECT_NE(nullptr, outMaxValue); in HWTEST_F()
319 tensor->SetBuffer(outMaxValue, sizeof(int64_t)); in HWTEST_F()
384 * @tc.desc: Verify the invalid param to argmax's outMaxValue of the build function
463 * @tc.desc: Verify the argmax without set outMaxValue of the build function
|
| /foundation/ai/neural_network_runtime/test/unittest/components/v2_0/inner_model/ |
| H A D | inner_model_test.cpp | 1092 bool outMaxValue {false}; in HWTEST_F()
1093 void* primitive = mindspore::lite::MindIR_ArgMaxFusion_CreatePrimitive(axis, topK, keepDims, outMaxValue); in HWTEST_F()
|
| /foundation/ai/neural_network_runtime/test/unittest/components/v1_0/device_manager/ |
| H A D | device_manager_test.cpp | 235 bool outMaxValue {false}; in HWTEST_F()
236 void* primitive = mindspore::lite::MindIR_ArgMaxFusion_CreatePrimitive(axis, topK, keepDims, outMaxValue); in HWTEST_F()
|
| /foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ |
| H A D | lite_graph_to_hdi_model_v2_0.cpp | 83 argMaxFusion.outMaxValue = mindspore::lite::MindIR_ArgMaxFusion_GetOutMaxValue(primitive); in ConvertArgMaxFusion()
|
| H A D | lite_graph_to_hdi_model_v1_0.cpp | 83 argMaxFusion.outMaxValue = mindspore::lite::MindIR_ArgMaxFusion_GetOutMaxValue(primitive); in ConvertArgMaxFusion()
|
| H A D | lite_graph_to_hdi_model_v2_1.cpp | 100 argMaxFusion.outMaxValue = mindspore::lite::MindIR_ArgMaxFusion_GetOutMaxValue(primitive);
in ConvertArgMaxFusion()
|
| /foundation/ai/neural_network_runtime/test/unittest/components/v1_0/inner_model/ |
| H A D | inner_model_test.cpp | 1153 bool outMaxValue {false}; in HWTEST_F()
1154 void* primitive = mindspore::lite::MindIR_ArgMaxFusion_CreatePrimitive(axis, topK, keepDims, outMaxValue); in HWTEST_F()
|
Completed in 20 milliseconds