Searched refs:TopKBuilderTest (Results 1 - 1 of 1) sorted by relevance
/foundation/ai/neural_network_runtime/test/unittest/ops/ |
H A D | topk_builder_test.cpp | 29 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