Home
last modified time | relevance | path

Searched refs:axisReturn (Results 1 - 5 of 5) sorted by relevance

/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dunsqueeze_builder_test.cpp293 auto axisReturn = mindspore::lite::MindIR_Unsqueeze_GetAxis(primitive.get()); in HWTEST_F() local
294 auto axisReturnSize = axisReturn.size(); in HWTEST_F()
296 EXPECT_EQ(axisReturn[i], m_expectAxisValue[i]); in HWTEST_F()
H A Dtopk_builder_test.cpp331 auto axisReturn = mindspore::lite::MindIR_TopKFusion_GetAxis(primitive.get()); in HWTEST_F() local
332 EXPECT_EQ(axisReturn, axisValue); in HWTEST_F()
H A Dquant_dtype_cast_builder_test.cpp403 auto axisReturn = mindspore::lite::MindIR_QuantDTypeCast_GetAxis(primitive.get()); in HWTEST_F() local
404 EXPECT_EQ(axisReturn, axisValue); in HWTEST_F()
H A Dscale_builder_test.cpp370 auto axisReturn = mindspore::lite::MindIR_ScaleFusion_GetAxis(primitive.get()); in HWTEST_F() local
371 EXPECT_EQ(axisReturn, axisValue); in HWTEST_F()
H A Dargmax_test.cpp503 int64_t axisReturn = mindspore::lite::MindIR_ArgMaxFusion_GetAxis(primitive.get()); in HWTEST_F() local
504 EXPECT_EQ(axisReturn, 0); in HWTEST_F()

Completed in 4 milliseconds