Home
last modified time | relevance | path

Searched refs:tensorDescImpl (Results 1 – 4 of 4) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_core/
H A Dneural_network_core.cpp804 if (tensorDescImpl == nullptr) { in OH_NNTensorDesc_Create()
825 delete tensorDescImpl; in OH_NNTensorDesc_Destroy()
842 return tensorDescImpl->SetName(name); in OH_NNTensorDesc_SetName()
861 return tensorDescImpl->GetName(name); in OH_NNTensorDesc_GetName()
872 return tensorDescImpl->SetDataType(dataType); in OH_NNTensorDesc_SetDataType()
887 return tensorDescImpl->GetDataType(dataType); in OH_NNTensorDesc_GetDataType()
941 return tensorDescImpl->SetFormat(format); in OH_NNTensorDesc_SetFormat()
956 return tensorDescImpl->GetFormat(format); in OH_NNTensorDesc_GetFormat()
986 return tensorDescImpl->GetByteSize(byteSize); in OH_NNTensorDesc_GetByteSize()
1147 auto tensorDescImpl = tensorImpl->GetTensorDesc(); in OH_NNTensor_GetTensorDesc() local
[all …]
/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/
H A Dnnexecutor.cpp155 TensorDesc* tensorDescImpl = new (std::nothrow) TensorDesc(); in CreateInputTensorDesc() local
156 if (tensorDescImpl == nullptr) { in CreateInputTensorDesc()
162 *tensorDescImpl = *(m_inputTensorDescs[index].first.get()); in CreateInputTensorDesc()
164 return reinterpret_cast<NN_TensorDesc*>(tensorDescImpl); in CreateInputTensorDesc()
178 TensorDesc* tensorDescImpl = new (std::nothrow) TensorDesc(); in CreateOutputTensorDesc() local
179 if (tensorDescImpl == nullptr) { in CreateOutputTensorDesc()
185 *tensorDescImpl = *(m_outputTensorDescs[index].first.get()); in CreateOutputTensorDesc()
187 return reinterpret_cast<NN_TensorDesc*>(tensorDescImpl); in CreateOutputTensorDesc()
H A Dnn_tensor.cpp166 …const auto* tensorDescImpl = reinterpret_cast<const OHOS::NeuralNetworkRuntime::TensorDesc*>(tenso… in BuildFromTensorDesc() local
170 OH_NN_ReturnCode returnCode = tensorDescImpl->GetDataType(&dataType); in BuildFromTensorDesc()
183 returnCode = tensorDescImpl->GetShape(&shape, &shapeNum); in BuildFromTensorDesc()
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/neural_network_runtime_test/
H A Dneural_network_runtime_test.cpp2947 TensorDesc* tensorDescImpl = new (std::nothrow) TensorDesc(); variable
2948 NN_TensorDesc* tensor = reinterpret_cast<NN_TensorDesc*>(tensorDescImpl);
2975 TensorDesc* tensorDescImpl = new (std::nothrow) TensorDesc(); variable
2976 NN_TensorDesc* tensor = reinterpret_cast<NN_TensorDesc*>(tensorDescImpl);