| /ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/inner_model/ |
| H A D | nn_tensor_test.cpp | 42 NNTensor nnTensor; variable
60 NNTensor nnTensor; variable
74 NNTensor nnTensor; variable
88 NNTensor nnTensor; variable
102 NNTensor nnTensor; variable
119 NNTensor nnTensor; variable
138 NNTensor nnTensor; variable
156 NNTensor nnTensor; variable
174 NNTensor nnTensor; variable
192 NNTensor nnTensor; variable
[all …]
|
| /ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v2_0/inner_model/ |
| H A D | nn_tensor_test.cpp | 42 NNTensor nnTensor; variable
60 NNTensor nnTensor; variable
74 NNTensor nnTensor; variable
88 NNTensor nnTensor; variable
102 NNTensor nnTensor; variable
119 NNTensor nnTensor; variable
138 NNTensor nnTensor; variable
156 NNTensor nnTensor; variable
174 NNTensor nnTensor; variable
192 NNTensor nnTensor; variable
[all …]
|
| /ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/nn_tensor/ |
| H A D | nn_tensor_test.cpp | 131 delete nnTensor;
147 nnTensor->SetSize(1);
151 nnTensor->SetFd(-1);
152 delete nnTensor;
168 nnTensor->SetSize(1);
172 nnTensor->SetFd(0);
173 delete nnTensor;
1114 nnTensor->SetSize(200);
1115 nnTensor->SetOffset(0);
1145 nnTensor->SetSize(200);
[all …]
|
| /ohos5.0/foundation/ai/neural_network_runtime/example/deep_learning_framework/tflite/delegates/nnrt_delegate/ |
| H A D | nnrt_op_builder.cpp | 354 OH_NN_Tensor nnTensor; in AddTensor() local
367 …TF_LITE_ENSURE_STATUS(GetTensorSize(m_context, nnTensor.dimensions, nnTensor.dimensionCount, tenso… in AddTensor()
378 int32_t nnRet = m_nnrt->OH_NNModel_AddTensor(m_nnModel, &nnTensor); in AddTensor()
401 const int32_t* tensorDims = nnTensor.dimensions; in TransposeDepthwiseTensor()
402 uint32_t tensorRank = nnTensor.dimensionCount; in TransposeDepthwiseTensor()
406 nnTensor.dimensions = weightDims.data(); in TransposeDepthwiseTensor()
430 OH_NN_QuantParam& nnQuantParam, OH_NN_Tensor& nnTensor) in ConstructNNTensor() argument
456 nnTensor.dataType = nnType; in ConstructNNTensor()
457 nnTensor.dimensionCount = tensorRank; in ConstructNNTensor()
458 nnTensor.dimensions = tensorDims; in ConstructNNTensor()
[all …]
|
| H A D | nnrt_delegate_kernel.cpp | 290 …nst std::pair<int32_t, int32_t>& indexPair, OH_NN_QuantParam* nnQuantParam, OH_NN_Tensor& nnTensor) in ConvertTensorTypeToNn() argument
313 nnTensor.dataType = nnType; in ConvertTensorTypeToNn()
314 nnTensor.dimensionCount = tensorRank; in ConvertTensorTypeToNn()
315 nnTensor.dimensions = tensorDims; in ConvertTensorTypeToNn()
316 nnTensor.quantParam = nnQuantParam; in ConvertTensorTypeToNn()
317 nnTensor.type = OH_NN_TENSOR; in ConvertTensorTypeToNn()
323 OH_NNExecutor* pNnExecution, OH_NN_Tensor& nnTensor) in SetInputTensors() argument
341 ConvertTensorTypeToNn(context, indexPair, nnQuantParam, nnTensor); in SetInputTensors()
348 &nnTensor, tensor->data.raw, tensor->bytes), in SetInputTensors()
|
| H A D | nnrt_delegate_kernel.h | 71 OH_NN_QuantParam* nnQuantParam, OH_NN_Tensor& nnTensor);
73 OH_NN_Tensor& nnTensor);
|
| H A D | nnrt_op_builder.h | 281 …TfLiteStatus TransposeDepthwiseTensor(int32_t tensorIndex, OH_NN_Tensor& nnTensor, std::vector<int…
286 OH_NN_QuantParam& nnQuantParam, OH_NN_Tensor& nnTensor);
|
| /ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ |
| H A D | inner_model.cpp | 58 std::shared_ptr<NNTensor> nnTensor = CreateSharedPtr<NNTensor>(); in ConstructNNTensorFromLiteGraphTensor() local
59 if (nnTensor == nullptr) { in ConstructNNTensorFromLiteGraphTensor()
70 nnTensor->SetFormat(nnFormat); in ConstructNNTensorFromLiteGraphTensor()
72 return nnTensor; in ConstructNNTensorFromLiteGraphTensor()
90 std::shared_ptr<NNTensor> nnTensor; in ConstructNNTensorsFromLiteGraph() local
93 if (nnTensor == nullptr) { in ConstructNNTensorsFromLiteGraph()
98 nnTensors.emplace_back(nnTensor); in ConstructNNTensorsFromLiteGraph()
176 OH_NN_ReturnCode InnerModel::AddTensor(const OH_NN_Tensor& nnTensor) in AddTensor() argument
189 OH_NN_ReturnCode ret = tensor->BuildFromOHNNTensor(nnTensor); in AddTensor()
622 if (nnTensor->IsOpParameter()) { in AddTensorsToLiteGraph()
[all …]
|
| H A D | nn_tensor.cpp | 105 OH_NN_ReturnCode NNTensor::BuildFromOHNNTensor(const OH_NN_Tensor& nnTensor) in BuildFromOHNNTensor() argument
107 m_type = nnTensor.type; in BuildFromOHNNTensor()
109 if (!Validation::ValidateTensorDataType(nnTensor.dataType)) { in BuildFromOHNNTensor()
110 LOGE("BuildFromOHNNTensor failed, passed invalid data type: %d.", nnTensor.dataType); in BuildFromOHNNTensor()
113 m_dataType = nnTensor.dataType; in BuildFromOHNNTensor()
115 if (!Validation::ValidateTensorType(nnTensor.type)) { in BuildFromOHNNTensor()
116 LOGE("BuildFromOHNNTensor failed, passed invalid nnTensor type: %d.", nnTensor.type); in BuildFromOHNNTensor()
120 OH_NN_ReturnCode returnCode = ParseDimensions(nnTensor.dimensions, nnTensor.dimensionCount); in BuildFromOHNNTensor()
126 returnCode = ParseQuantParams(nnTensor.quantParam); in BuildFromOHNNTensor()
|
| H A D | hdi_prepared_model_v1_0.cpp | 103 const NNTensor2_0* nnTensor = reinterpret_cast<const NNTensor2_0*>(tensor); in TransIOTensor() local
104 TensorDesc* nnTensorDesc = nnTensor->GetTensorDesc(); in TransIOTensor()
151 if (!nnTensor->CheckTensorData()) { in TransIOTensor()
155 …V1_0::SharedBuffer iBuffer {nnTensor->GetFd(), nnTensor->GetSize(), nnTensor->GetOffset(), nnTenso… in TransIOTensor()
|
| H A D | hdi_prepared_model_v2_0.cpp | 104 const NNTensor2_0* nnTensor = reinterpret_cast<const NNTensor2_0*>(tensor); in TransIOTensor() local
105 TensorDesc* nnTensorDesc = nnTensor->GetTensorDesc(); in TransIOTensor()
152 if (!nnTensor->CheckTensorData()) { in TransIOTensor()
156 …V2_0::SharedBuffer iBuffer {nnTensor->GetFd(), nnTensor->GetSize(), nnTensor->GetOffset(), nnTenso… in TransIOTensor()
|
| H A D | hdi_prepared_model_v2_1.cpp | 104 const NNTensor2_0* nnTensor = reinterpret_cast<const NNTensor2_0*>(tensor); in TransIOTensor() local
105 TensorDesc* nnTensorDesc = nnTensor->GetTensorDesc(); in TransIOTensor()
152 if (!nnTensor->CheckTensorData()) { in TransIOTensor()
156 …V2_1::SharedBuffer iBuffer {nnTensor->GetFd(), nnTensor->GetSize(), nnTensor->GetOffset(), nnTenso… in TransIOTensor()
|
| H A D | nnexecutor.cpp | 297 TensorDesc* nnTensorDesc = nnTensor->GetTensorDesc(); in RunSync()
363 const NNTensor2_0* nnTensor = nullptr; in CheckInputDimRanges() local
368 nnTensor = reinterpret_cast<const NNTensor2_0*>(inputTensors[i]); in CheckInputDimRanges()
369 if (nnTensor == nullptr) { in CheckInputDimRanges()
441 auto ret = inputTensor->BuildFromOHNNTensor(nnTensor); in BuildInputTensor()
546 …std::vector<int32_t> tensorShape = ConstructVectorFromArray(nnTensor.dimensions, nnTensor.dimensio… in CheckInputDimRanges()
573 auto nnRet = CheckInputDimRanges(index, nnTensor); in SetInput()
587 auto ret = BuildInputTensor(index, nnTensor, inputTensor); in SetInput()
644 uint32_t index, const OH_NN_Tensor& nnTensor, const OH_NN_Memory& memory) in SetInputFromMemory() argument
646 auto nnRet = CheckInputDimRanges(index, nnTensor); in SetInputFromMemory()
[all …]
|
| H A D | nnexecutor.h | 64 …OH_NN_ReturnCode SetInput(uint32_t index, const OH_NN_Tensor& nnTensor, const void* buffer, size_t…
65 …OH_NN_ReturnCode SetInputFromMemory(uint32_t index, const OH_NN_Tensor& nnTensor, const OH_NN_Memo…
87 OH_NN_ReturnCode BuildInputTensor(uint32_t index, const OH_NN_Tensor& nnTensor,
93 OH_NN_ReturnCode CheckInputDimRanges(uint32_t index, const OH_NN_Tensor& nnTensor) const;
|
| H A D | nn_tensor.h | 43 OH_NN_ReturnCode BuildFromOHNNTensor(const OH_NN_Tensor& nnTensor);
|
| H A D | inner_model.h | 38 OH_NN_ReturnCode AddTensor(const OH_NN_Tensor& nnTensor);
|
| /ohos5.0/foundation/ai/neural_network_runtime/test/unittest/common/ |
| H A D | base_test.cpp | 31 std::shared_ptr<NNTensor> nnTensor = std::make_shared<NNTensor>(); in TransToNNTensor() local
38 nnTensor->BuildFromOHNNTensor(tensor); in TransToNNTensor()
39 return nnTensor; in TransToNNTensor()
|
| /ohos5.0/foundation/ai/neural_network_runtime/example/deep_learning_framework/tflite/nnrt/ |
| H A D | nnrt_implementation.h | 38 OH_NN_ReturnCode (*OH_NNModel_AddTensor)(OH_NNModel* model, const OH_NN_Tensor* nnTensor);
64 const OH_NN_Tensor* nnTensor, const void* buffer, size_t length);
75 const OH_NN_Tensor* nnTensor, const OH_NN_Memory* memory);
|
| /ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/hdi_prepared_model/ |
| H A D | hdi_prepared_model_test.cpp | 741 NNTensor2_0* nnTensor = new (std::nothrow) NNTensor2_0(backendId); variable
742 EXPECT_NE(nullptr, nnTensor);
754 OH_NN_ReturnCode retSetTensorDesc = nnTensor->SetTensorDesc(&tensorDesc);
757 nnTensor->SetSize(200);
758 nnTensor->SetOffset(0);
761 nnTensor->SetData(buffer);
763 NN_Tensor* tensor = reinterpret_cast<NN_Tensor*>(nnTensor);
|
| /ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v2_0/hdi_prepared_model/ |
| H A D | hdi_prepared_model_test.cpp | 736 NNTensor2_0* nnTensor = new (std::nothrow) NNTensor2_0(backendId); variable
737 EXPECT_NE(nullptr, nnTensor);
749 OH_NN_ReturnCode retSetTensorDesc = nnTensor->SetTensorDesc(&tensorDesc);
752 nnTensor->SetSize(200);
753 nnTensor->SetOffset(0);
756 nnTensor->SetData(buffer);
758 NN_Tensor* tensor = reinterpret_cast<NN_Tensor*>(nnTensor);
|
| /ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v2_1/hdi_prepared_model/ |
| H A D | hdi_prepared_model_test.cpp | 735 NNTensor2_0* nnTensor = new (std::nothrow) NNTensor2_0(backendId); variable
736 EXPECT_NE(nullptr, nnTensor);
748 OH_NN_ReturnCode retSetTensorDesc = nnTensor->SetTensorDesc(&tensorDesc);
751 nnTensor->SetSize(200);
752 nnTensor->SetOffset(0);
755 nnTensor->SetData(buffer);
757 NN_Tensor* tensor = reinterpret_cast<NN_Tensor*>(nnTensor);
|