Searched refs:ArgMaxBuilderTest (Results 1 – 1 of 1) sorted by relevance
| /ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/ |
| H A D | argmax_test.cpp | 27 class ArgMaxBuilderTest : public OpsTest { class
51 void ArgMaxBuilderTest::SetUp() {} in SetUp()
53 void ArgMaxBuilderTest::TearDown() {} in TearDown()
55 void ArgMaxBuilderTest::SetArgmaxAxis(OH_NN_DataType dataType, in SetArgmaxAxis()
75 void ArgMaxBuilderTest::SetArgmaxTopK(OH_NN_DataType dataType, in SetArgmaxTopK()
100 HWTEST_F(ArgMaxBuilderTest, argmax_build_001, TestSize.Level1)
118 HWTEST_F(ArgMaxBuilderTest, argmax_build_002, TestSize.Level1)
142 HWTEST_F(ArgMaxBuilderTest, argmax_build_003, TestSize.Level1)
165 HWTEST_F(ArgMaxBuilderTest, argmax_build_004, TestSize.Level1)
188 HWTEST_F(ArgMaxBuilderTest, argmax_build_005, TestSize.Level1)
[all …]
|