Home
last modified time | relevance | path

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

/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dtopk_builder_test.cpp65 std::shared_ptr<NNTensor> topkTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveSortedTensor() local
68 topkTensor->SetBuffer(topkValue, sizeof(bool)); in SaveSortedTensor()
69 m_allTensors.emplace_back(topkTensor); in SaveSortedTensor()
223 std::shared_ptr<NNTensor> topkTensor = TransToNNTensor(OH_NN_BOOL, paramDim, nullptr, OH_NN_TOP_K_SORTED); in HWTEST_F() local
224 topkTensor->SetBuffer(nullptr, 0); in HWTEST_F()
225 m_allTensors.emplace_back(topkTensor); in HWTEST_F()

Completed in 2 milliseconds