Home
last modified time | relevance | path

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

/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Ddepthwise_conv2d_native_pad_test.cpp81 std::shared_ptr<NNTensor> inputsTensor; in SetDepthwiseConv2dInput() local
83 m_allTensors.emplace_back(inputsTensor); in SetDepthwiseConv2dInput()
88 inputsTensor->SetBuffer(weightValue, weightNum * sizeof(weightValue)); in SetDepthwiseConv2dInput()
89 m_allTensors.emplace_back(inputsTensor); 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; variable
433 m_allTensors.emplace_back(inputsTensor);
440 m_allTensors.emplace_back(inputsTensor);
446 inputsTensor->SetBuffer(biasValue, biasNum * sizeof(float));
[all …]
H A Dconv2d_pad_test.cpp79 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()