Searched defs:tensorDesc (Results 1 – 6 of 6) sorted by relevance
| /foundation/ai/neural_network_runtime/frameworks/native/neural_network_core/ |
| D | neural_network_core.cpp | 569 NN_TensorDesc *tensorDesc = reinterpret_cast<NN_TensorDesc *>(tensorDescImpl); in OH_NNTensorDesc_Create() local
573 NNRT_API OH_NN_ReturnCode OH_NNTensorDesc_Destroy(NN_TensorDesc **tensorDesc) in OH_NNTensorDesc_Destroy()
590 NNRT_API OH_NN_ReturnCode OH_NNTensorDesc_SetName(NN_TensorDesc *tensorDesc, const char *name) in OH_NNTensorDesc_SetName()
605 NNRT_API OH_NN_ReturnCode OH_NNTensorDesc_GetName(const NN_TensorDesc *tensorDesc, const char **nam… in OH_NNTensorDesc_GetName()
624 NNRT_API OH_NN_ReturnCode OH_NNTensorDesc_SetDataType(NN_TensorDesc *tensorDesc, OH_NN_DataType dat… in OH_NNTensorDesc_SetDataType()
635 NNRT_API OH_NN_ReturnCode OH_NNTensorDesc_GetDataType(const NN_TensorDesc *tensorDesc, OH_NN_DataTy… in OH_NNTensorDesc_GetDataType()
650 NNRT_API OH_NN_ReturnCode OH_NNTensorDesc_SetShape(NN_TensorDesc *tensorDesc, const int32_t *shape,… in OH_NNTensorDesc_SetShape()
668 NNRT_API OH_NN_ReturnCode OH_NNTensorDesc_GetShape(const NN_TensorDesc *tensorDesc, in OH_NNTensorDesc_GetShape()
693 NNRT_API OH_NN_ReturnCode OH_NNTensorDesc_SetFormat(NN_TensorDesc *tensorDesc, OH_NN_Format format) in OH_NNTensorDesc_SetFormat()
704 NNRT_API OH_NN_ReturnCode OH_NNTensorDesc_GetFormat(const NN_TensorDesc *tensorDesc, OH_NN_Format *… in OH_NNTensorDesc_GetFormat()
[all …]
|
| /foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ |
| D | nncompiler.cpp | 672 for (const auto& tensorDesc : tensorDescs) { in SerializeTensorsToBuffer() local
685 for (const auto& tensorDesc : immediateTensorDescs) { in SerializeTensorsToBuffer() local
703 for (const auto& tensorDesc : immediateTensorDescs) { in SerializeTensorsToBuffer() local
|
| D | nntensor.cpp | 43 OH_NN_ReturnCode NNTensor2_0::SetTensorDesc(const TensorDesc* tensorDesc) in SetTensorDesc()
|
| D | nnexecutor.cpp | 95 auto tensorDesc = m_outputTensorDescs[outputIndex].first; in GetOutputShape() local
308 …const std::pair<std::shared_ptr<TensorDesc>, OH_NN_TensorType>& tensorDesc, const NNTensor& tensor… in CompareAttribute()
605 const std::pair<std::shared_ptr<TensorDesc>, OH_NN_TensorType>& tensorDesc) in BuildNNTensorFromDesc()
|
| D | neural_network_runtime.cpp | 134 OH_NN_ReturnCode OH_NNModel_AddTensorToModel(OH_NNModel* model, const NN_TensorDesc* tensorDesc) in OH_NNModel_AddTensorToModel()
|
| D | nn_tensor.cpp | 159 OH_NN_ReturnCode NNTensor::BuildFromTensorDesc(const NN_TensorDesc* tensorDesc) in BuildFromTensorDesc()
|