Home
last modified time | relevance | path

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

/ohos5.0/foundation/ai/neural_network_runtime/example/deep_learning_framework/tflite/delegates/nnrt_delegate/
H A Dnnrt_op_builder.cpp59 TfLiteTensor* biasTensor = &mappingArgs.context->tensors[biasIndex]; in AddZerosBias() local
63 biasTensor->type = kTfLiteFloat32; in AddZerosBias()
65 biasTensor->type = kTfLiteInt32; in AddZerosBias()
71 biasTensor->allocation_type = kTfLiteDynamic; in AddZerosBias()
72 mappingArgs.context->ResizeTensor(mappingArgs.context, biasTensor, biasShape); in AddZerosBias()
77 memset_s(biasTensor->data.f, biasTensor->bytes, 0, channelNum * sizeof(float)); in AddZerosBias()
78 … TF_LITE_ENSURE_STATUS(mappingArgs.builder->AddVectorFloat32Tensor(biasTensor->data.f, channelNum, in AddZerosBias()
81 memset_s(biasTensor->data.i32, biasTensor->bytes, 0, channelNum * sizeof(int32_t)); in AddZerosBias()
86 biasTensor->params.scale = inputTensor.params.scale * filterTensor.params.scale; in AddZerosBias()
87 … TF_LITE_ENSURE_STATUS(mappingArgs.builder->AddVectorInt32Tensor(biasTensor->data.i32, channelNum, in AddZerosBias()
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dlrn_test.cpp91 std::shared_ptr<NNTensor> biasTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveBias() local
94 biasTensor->SetBuffer(biasValue, sizeof(float)); in SaveBias()
95 m_allTensors.emplace_back(biasTensor); in SaveBias()
301 std::shared_ptr<NNTensor> biasTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, variable
304 biasTensor->SetBuffer(biasValue, sizeof(int64_t));
305 m_allTensors.emplace_back(biasTensor);
510 std::shared_ptr<NNTensor> biasTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim, variable
512 m_allTensors.emplace_back(biasTensor);
H A Dinstance_norm_test.cpp73 std::shared_ptr<NNTensor> biasTensor; in SetInputTensor() local
74 biasTensor = TransToNNTensor(OH_NN_FLOAT32, m_scaleAndBiasDim, nullptr, OH_NN_TENSOR); in SetInputTensor()
75 m_allTensors.emplace_back(biasTensor); in SetInputTensor()