Home
last modified time | relevance | path

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

/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/
H A Dinner_model.cpp721 std::pair<std::shared_ptr<TensorDesc>, OH_NN_TensorType> tensorDescPair; in GetInputTensorDescs() local
723 tensorDescPair.first = OHOS::NeuralNetworkRuntime::CreateSharedPtr<TensorDesc>(); in GetInputTensorDescs()
724 inputTensor->ConvertToTensorDesc(*(tensorDescPair.first.get())); in GetInputTensorDescs()
725 tensorDescPair.second = inputTensor->GetType(); in GetInputTensorDescs()
726 inputTensorDescs.emplace_back(tensorDescPair); in GetInputTensorDescs()
735 std::pair<std::shared_ptr<TensorDesc>, OH_NN_TensorType> tensorDescPair; in GetOutputTensorDescs() local
737 tensorDescPair.first = OHOS::NeuralNetworkRuntime::CreateSharedPtr<TensorDesc>(); in GetOutputTensorDescs()
738 outputTensor->ConvertToTensorDesc(*(tensorDescPair.first.get())); in GetOutputTensorDescs()
739 tensorDescPair.second = outputTensor->GetType(); in GetOutputTensorDescs()
740 outputTensorDescs.emplace_back(tensorDescPair); in GetOutputTensorDescs()
H A Dnncompiler.cpp944 std::pair<std::shared_ptr<TensorDesc>, OH_NN_TensorType> tensorDescPair; in DeserializedTensorsFromBuffer() local
945 tensorDescPair.first = CreateSharedPtr<TensorDesc>(); in DeserializedTensorsFromBuffer()
946 if (tensorDescPair.first == nullptr) { in DeserializedTensorsFromBuffer()
952 ret = immediateTensorDesc.CopyToTensorDesc(*(tensorDescPair.first.get())); in DeserializedTensorsFromBuffer()
960 tensorDescPair.second = immediateTensorDesc.m_tensorType; in DeserializedTensorsFromBuffer()
962 tensorDescs.emplace_back(tensorDescPair); in DeserializedTensorsFromBuffer()