Searched refs:inputsTensor (Results 1 - 2 of 2) sorted by relevance
/foundation/ai/neural_network_runtime/test/unittest/ops/ |
H A D | depthwise_conv2d_native_pad_test.cpp | 81 std::shared_ptr<NNTensor> inputsTensor; in SetDepthwiseConv2dInput() local 82 inputsTensor = TransToNNTensor(OH_NN_FLOAT32, m_input_dim, nullptr, OH_NN_TENSOR); in SetDepthwiseConv2dInput() 83 m_allTensors.emplace_back(inputsTensor); in SetDepthwiseConv2dInput() 84 inputsTensor = TransToNNTensor(OH_NN_FLOAT32, weightDim, nullptr, OH_NN_TENSOR); in SetDepthwiseConv2dInput() 88 inputsTensor->SetBuffer(weightValue, weightNum * sizeof(weightValue)); in SetDepthwiseConv2dInput() 89 m_allTensors.emplace_back(inputsTensor); in SetDepthwiseConv2dInput() 90 inputsTensor = TransToNNTensor(OH_NN_FLOAT32, biasDim, nullptr, OH_NN_TENSOR); in SetDepthwiseConv2dInput() 93 inputsTensor->SetBuffer(biasValue, biasNum * sizeof(float)); in SetDepthwiseConv2dInput() 94 m_allTensors.emplace_back(inputsTensor); in SetDepthwiseConv2dInput() 431 std::shared_ptr<NNTensor> inputsTensor; in HWTEST_F() local [all...] |
H A D | conv2d_pad_test.cpp | 79 std::shared_ptr<NNTensor> inputsTensor; in SetConv2dInput() local 80 inputsTensor = TransToNNTensor(OH_NN_FLOAT32, m_input_dim, nullptr, OH_NN_TENSOR); in SetConv2dInput() 81 m_allTensors.emplace_back(inputsTensor); in SetConv2dInput() 83 inputsTensor = TransToNNTensor(OH_NN_FLOAT32, weightDim, nullptr, OH_NN_TENSOR); in SetConv2dInput() 87 inputsTensor->SetBuffer(weightValue, weightNum * sizeof(weightValue)); in SetConv2dInput() 88 m_allTensors.emplace_back(inputsTensor); in SetConv2dInput() 89 inputsTensor = TransToNNTensor(OH_NN_FLOAT32, biasDim, nullptr, OH_NN_TENSOR); in SetConv2dInput() 93 inputsTensor->SetBuffer(biasValue, sizeof(float)); in SetConv2dInput() 94 m_allTensors.emplace_back(inputsTensor); in SetConv2dInput()
|
Completed in 4 milliseconds