Home
last modified time | relevance | path

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

/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Ddetection_post_process_test.cpp75 std::shared_ptr<NNTensor> inputSizeTensor = TransToNNTensor(dataType, dim, quantParam, type); in SetInputSize() local
78 inputSizeTensor->SetBuffer(inputSizeValue, sizeof(int64_t)); in SetInputSize()
79 m_allTensors.emplace_back(inputSizeTensor); in SetInputSize()
328 std::shared_ptr<NNTensor> inputSizeTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim, variable
331 inputSizeTensor->SetBuffer(inputSizeValue, sizeof(float));
332 m_allTensors.emplace_back(inputSizeTensor);
870 std::shared_ptr<NNTensor> inputSizeTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, variable
872 m_allTensors.emplace_back(inputSizeTensor);
H A Dlstm_test.cpp97 std::shared_ptr<NNTensor> inputSizeTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveInputSize() local
100 inputSizeTensor->SetBuffer(inputSizeValue, sizeof(int64_t)); in SaveInputSize()
101 m_allTensors.emplace_back(inputSizeTensor); in SaveInputSize()
426 std::shared_ptr<NNTensor> inputSizeTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim, variable
430 inputSizeTensor->SetBuffer(inputSizeValue, sizeof(float));
431 m_allTensors.emplace_back(inputSizeTensor);
995 std::shared_ptr<NNTensor> inputSizeTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, variable
997 m_allTensors.emplace_back(inputSizeTensor);