Home
last modified time | relevance | path

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

/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dtopk_builder_test.cpp29 class TopKBuilderTest : public OpsTest { class
43 void TopKBuilderTest::InitTensor(const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex) in InitTensor()
52 void TopKBuilderTest::SaveAxisTensor(OH_NN_DataType dataType, const std::vector<int32_t> &dim, in SaveAxisTensor()
62 void TopKBuilderTest::SaveSortedTensor(OH_NN_DataType dataType, const std::vector<int32_t> &dim, in SaveSortedTensor()
78 HWTEST_F(TopKBuilderTest, topk_build_001, TestSize.Level0) in HWTEST_F()
98 HWTEST_F(TopKBuilderTest, topk_build_002, TestSize.Level0) in HWTEST_F()
119 HWTEST_F(TopKBuilderTest, topk_build_003, TestSize.Level0) in HWTEST_F()
139 HWTEST_F(TopKBuilderTest, topk_builder_004, TestSize.Level0) in HWTEST_F()
159 HWTEST_F(TopKBuilderTest, topk_build_005, TestSize.Level0) in HWTEST_F()
174 HWTEST_F(TopKBuilderTest, topk_build_00
[all...]

Completed in 2 milliseconds