Home
last modified time | relevance | path

Searched refs:outputTensors (Results 1 – 8 of 8) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/
H A Dnnexecutor.cpp241 NN_Tensor* outputTensors[], size_t outputSize) in RunSync() argument
273 if (outputTensors[i] == nullptr) { in RunSync()
277 outputTensorsVec.emplace_back(outputTensors[i]); in RunSync()
971 std::vector<std::shared_ptr<NNTensor>>& outputTensors) in Run() argument
977 size_t outputSize = outputTensors.size(); in Run()
985 outputTensors[i]->ConvertToIOTensor(tensor); in Run()
1004 ret = outputTensors[i]->SetDimensions(outputsDims[i]); in Run()
1034 std::vector<std::shared_ptr<NNTensor>> outputTensors; in Run() local
1041 outputTensors.emplace_back(m_outputTensors[i].tensor); in Run()
1045 auto ret = Run(inputTensors, outputTensors); in Run()
[all …]
H A Dnnexecutor.h50 NN_Tensor* outputTensors[],
54 NN_Tensor* outputTensors[],
82 std::vector<std::shared_ptr<NNTensor>>& outputTensors);
/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_core/
H A Dexecutor.h51 NN_Tensor* outputTensors[],
55 NN_Tensor* outputTensors[],
/ohos5.0/foundation/ai/neural_network_runtime/example/deep_learning_framework/tflite/delegates/nnrt_delegate/
H A Dnnrt_delegate_kernel.cpp179 const TfLiteIntArray* inputTensors, const TfLiteIntArray* outputTensors) in BuildGraph() argument
187 TF_LITE_ENSURE_EQ(context, outputTensors != nullptr, true); in BuildGraph()
208 for (auto i : TfLiteIntArrayView(outputTensors)) { in BuildGraph()
H A Dnnrt_delegate_kernel.h69 const TfLiteIntArray* inputTensors, const TfLiteIntArray* outputTensors);
/ohos5.0/foundation/ai/neural_network_runtime/
H A Dneural-network-runtime-guidelines.md517 NN_Tensor* outputTensors[outputCount];
522 outputTensors[i] = tensor;
530 …returnCode = OH_NNExecutor_RunSync(executor, inputTensors, inputCount, outputTensors, outputCount);
534 Print(outputTensors, outputCount);
544 returnCode = OH_NNTensor_Destroy(&outputTensors[i]);
/ohos5.0/docs/zh-cn/application-dev/ai/nnrt/
H A Dneural-network-runtime-guidelines.md508 NN_Tensor* outputTensors[outputCount];
513 outputTensors[i] = tensor;
521 …returnCode = OH_NNExecutor_RunSync(executor, inputTensors, inputCount, outputTensors, outputCount);
525 Print(outputTensors, outputCount);
535 returnCode = OH_NNTensor_Destroy(&outputTensors[i]);
/ohos5.0/docs/en/application-dev/ai/nnrt/
H A Dneural-network-runtime-guidelines.md508 NN_Tensor* outputTensors[outputCount];
513 outputTensors[i] = tensor;
521 …returnCode = OH_NNExecutor_RunSync(executor, inputTensors, inputCount, outputTensors, outputCount);
525 Print(outputTensors, outputCount);
535 returnCode = OH_NNTensor_Destroy(&outputTensors[i]);