Home
last modified time | relevance | path

Searched refs:AddTensor (Results 1 – 13 of 13) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/example/deep_learning_framework/tflite/delegates/nnrt_delegate/
H A Dnnrt_op_builder.h101 return AddTensor(tensorIndex, builtinCode, m_augmentedInputs, tensorFlags);
106 return AddTensor(tensorIndex, builtinCode, m_augmentedOutputs, tensorFlags);
273 TfLiteStatus AddTensor(int32_t tensorIndex, int32_t builtinCode, std::vector<uint32_t>& indices,
278 …TfLiteStatus AddTensor(int32_t tensorIndex, int32_t builtinCode, int32_t tensorFlags, int32_t& nnT…
H A Dnnrt_op_builder.cpp326 TfLiteStatus NnrtOpBuilder::AddTensor(int32_t tensorIndex, int32_t builtinCode, std::vector<uint32_… in AddTensor() function in tflite::delegate::nnrt::NnrtOpBuilder
342 TF_LITE_ENSURE_STATUS(AddTensor(tensorIndex, builtinCode, tensorFlags, nnTensorIndex)); in AddTensor()
349 TfLiteStatus NnrtOpBuilder::AddTensor(int32_t tensorIndex, int32_t builtinCode, int32_t tensorFlags, in AddTensor() function in tflite::delegate::nnrt::NnrtOpBuilder
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/inner_model/
H A Dinner_model_test.cpp90 EXPECT_EQ(OH_NN_SUCCESS, m_innerModelTest.AddTensor(tensor)); in SetTensors()
91 EXPECT_EQ(OH_NN_SUCCESS, m_innerModelTest.AddTensor(tensor)); in SetTensors()
92 EXPECT_EQ(OH_NN_SUCCESS, m_innerModelTest.AddTensor(tensor)); in SetTensors()
330 EXPECT_EQ(OH_NN_SUCCESS, m_innerModelTest.AddTensor(tensor));
348 EXPECT_EQ(OH_NN_SUCCESS, m_innerModelTest.AddTensor(tensor));
494 EXPECT_EQ(OH_NN_SUCCESS, m_innerModelTest.AddTensor(tensor));
495 EXPECT_EQ(OH_NN_SUCCESS, m_innerModelTest.AddTensor(tensor));
496 EXPECT_EQ(OH_NN_SUCCESS, m_innerModelTest.AddTensor(tensor));
683 EXPECT_EQ(OH_NN_SUCCESS, m_innerModelTest.AddTensor(tensor));
829 EXPECT_EQ(OH_NN_SUCCESS, m_innerModelTest.AddTensor(tensor));
[all …]
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v2_0/inner_model/
H A Dinner_model_test.cpp90 EXPECT_EQ(OH_NN_SUCCESS, m_innerModelTest.AddTensor(tensor)); in SetTensors()
91 EXPECT_EQ(OH_NN_SUCCESS, m_innerModelTest.AddTensor(tensor)); in SetTensors()
92 EXPECT_EQ(OH_NN_SUCCESS, m_innerModelTest.AddTensor(tensor)); in SetTensors()
272 EXPECT_EQ(OH_NN_SUCCESS, m_innerModelTest.AddTensor(tensor));
289 EXPECT_EQ(OH_NN_SUCCESS, m_innerModelTest.AddTensor(tensor));
434 EXPECT_EQ(OH_NN_SUCCESS, m_innerModelTest.AddTensor(tensor));
435 EXPECT_EQ(OH_NN_SUCCESS, m_innerModelTest.AddTensor(tensor));
436 EXPECT_EQ(OH_NN_SUCCESS, m_innerModelTest.AddTensor(tensor));
623 EXPECT_EQ(OH_NN_SUCCESS, m_innerModelTest.AddTensor(tensor));
769 EXPECT_EQ(OH_NN_SUCCESS, m_innerModelTest.AddTensor(tensor));
[all …]
/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/
H A Dinner_model.h38 OH_NN_ReturnCode AddTensor(const OH_NN_Tensor& nnTensor);
H A Dneural_network_runtime_compat.cpp42 return innerModel->AddTensor(*tensor); in OH_NNModel_AddTensor()
H A Dinner_model.cpp176 OH_NN_ReturnCode InnerModel::AddTensor(const OH_NN_Tensor& nnTensor) in AddTensor() function in OHOS::NeuralNetworkRuntime::InnerModel
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v2_0/neural_network_runtime_test/
H A Dneural_network_runtime_test.cpp109 OH_NN_ReturnCode ret = model.AddTensor(input1); in BuildModel()
116 ret = model.AddTensor(input2); in BuildModel()
125 ret = model.AddTensor(activation); in BuildModel()
139 ret = model.AddTensor(output); in BuildModel()
190 EXPECT_EQ(OH_NN_SUCCESS, innerModel.AddTensor(tensor)); in AddModelTensor()
191 EXPECT_EQ(OH_NN_SUCCESS, innerModel.AddTensor(tensor)); in AddModelTensor()
192 EXPECT_EQ(OH_NN_SUCCESS, innerModel.AddTensor(tensor)); in AddModelTensor()
195 EXPECT_EQ(OH_NN_SUCCESS, innerModel.AddTensor(tensorParam)); in AddModelTensor()
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/nn_compiler/
H A Dnn_compiler_test.cpp97 MOCK_METHOD1(AddTensor, OH_NN_ReturnCode(const OH_NN_Tensor&));
125 OH_NN_ReturnCode ret = innerModel.AddTensor(input1); in BuildModel()
132 ret = innerModel.AddTensor(input2); in BuildModel()
141 ret = innerModel.AddTensor(activation); in BuildModel()
155 ret = innerModel.AddTensor(output); in BuildModel()
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/compilation/
H A Dcompilation_test.cpp34 OH_NN_ReturnCode ret = innerModel.AddTensor(input1); in BuildModel()
41 ret = innerModel.AddTensor(input2); in BuildModel()
50 ret = innerModel.AddTensor(activation); in BuildModel()
64 ret = innerModel.AddTensor(output); in BuildModel()
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v2_0/compilation/
H A Dcompilation_test.cpp34 OH_NN_ReturnCode ret = innerModel.AddTensor(input1); in BuildModel()
41 ret = innerModel.AddTensor(input2); in BuildModel()
50 ret = innerModel.AddTensor(activation); in BuildModel()
64 ret = innerModel.AddTensor(output); in BuildModel()
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/neural_network_runtime_test/
H A Dneural_network_runtime_test.cpp110 OH_NN_ReturnCode ret = model.AddTensor(input1); in BuildModel()
117 ret = model.AddTensor(input2); in BuildModel()
126 ret = model.AddTensor(activation); in BuildModel()
140 ret = model.AddTensor(output); in BuildModel()
191 EXPECT_EQ(OH_NN_SUCCESS, innerModel.AddTensor(tensor)); in AddModelTensor()
192 EXPECT_EQ(OH_NN_SUCCESS, innerModel.AddTensor(tensor)); in AddModelTensor()
193 EXPECT_EQ(OH_NN_SUCCESS, innerModel.AddTensor(tensor)); in AddModelTensor()
196 EXPECT_EQ(OH_NN_SUCCESS, innerModel.AddTensor(tensorParam)); in AddModelTensor()
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/neural_network_core_test/
H A Dneural_network_core_test.cpp39 OH_NN_ReturnCode ret = model.AddTensor(input1); in BuildModel()
46 ret = model.AddTensor(input2); in BuildModel()
55 ret = model.AddTensor(activation); in BuildModel()
69 ret = model.AddTensor(output); in BuildModel()