Searched refs:inputSizeTensor (Results 1 - 2 of 2) sorted by relevance
| /foundation/ai/neural_network_runtime/test/unittest/ops/ |
| H A D | detection_post_process_test.cpp | 75 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,
in HWTEST_F() local
331 inputSizeTensor->SetBuffer(inputSizeValue, sizeof(float));
in HWTEST_F()
332 m_allTensors.emplace_back(inputSizeTensor);
in HWTEST_F()
870 std::shared_ptr<NNTensor> inputSizeTensor = TransToNNTensor(OH_NN_INT64, m_paramDim,
in HWTEST_F() local
872 m_allTensors.emplace_back(inputSizeTensor);
in HWTEST_F()
|
| H A D | lstm_test.cpp | 97 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,
in HWTEST_F() local
430 inputSizeTensor->SetBuffer(inputSizeValue, sizeof(float));
in HWTEST_F()
431 m_allTensors.emplace_back(inputSizeTensor);
in HWTEST_F()
995 std::shared_ptr<NNTensor> inputSizeTensor = TransToNNTensor(OH_NN_INT64, m_paramDim,
in HWTEST_F() local
997 m_allTensors.emplace_back(inputSizeTensor);
in HWTEST_F()
|
Completed in 4 milliseconds