Home
last modified time | relevance | path

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

/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dargmax_test.cpp27 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()
65 void ArgMaxBuilderTest::SetArgmaxKeepdims(OH_NN_DataType dataType, in SetArgmaxKeepdims()
75 void ArgMaxBuilderTest::SetArgmaxTopK(OH_NN_DataType dataType, in SetArgmaxTopK()
85 void ArgMaxBuilderTest::SetArgmaxOutMaxValue(OH_NN_DataType dataType, in SetArgmaxOutMaxValue()
100 HWTEST_F(ArgMaxBuilderTest, argmax_build_001, TestSize.Level1) in HWTEST_F()
118 HWTEST_F(ArgMaxBuilderTest, argmax_build_002, TestSize.Level1) in HWTEST_F()
142 HWTEST_F(ArgMaxBuilderTest, argmax_build_00
[all...]

Completed in 1 milliseconds