Home
last modified time | relevance | path

Searched refs:nmsScoreThresholdTensor (Results 1 – 1 of 1) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Ddetection_post_process_test.cpp106 …std::shared_ptr<NNTensor> nmsScoreThresholdTensor = TransToNNTensor(dataType, dim, quantParam, typ… in SetNmsScoreThreshold() local
109 nmsScoreThresholdTensor->SetBuffer(nmsScoreThresholdValue, sizeof(float)); in SetNmsScoreThreshold()
110 m_allTensors.emplace_back(nmsScoreThresholdTensor); in SetNmsScoreThreshold()
419 std::shared_ptr<NNTensor> nmsScoreThresholdTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, variable
422 nmsScoreThresholdTensor->SetBuffer(nmsScoreThresholdValue, sizeof(int64_t));
423 m_allTensors.emplace_back(nmsScoreThresholdTensor);
954 std::shared_ptr<NNTensor> nmsScoreThresholdTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim, variable
956 m_allTensors.emplace_back(nmsScoreThresholdTensor);