Home
last modified time | relevance | path

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

/ohos5.0/foundation/ai/neural_network_runtime/example/deep_learning_framework/tflite/delegates/nnrt_delegate/
H A Dtensor_mapping.h125 …nvertType(TfLiteContext* context, int32_t tensorIndex, int32_t tensorFlags, OH_NN_DataType& nnType) in ConvertType() argument
131 nnType = OH_NN_FLOAT32; in ConvertType()
133 nnType = OH_NN_FLOAT16; in ConvertType()
135 nnType = OH_NN_FLOAT32; in ConvertType()
139 nnType = OH_NN_INT32; in ConvertType()
141 nnType = OH_NN_INT8; in ConvertType()
143 nnType = (nnTypeEquivalent == kTfLiteInt32) ? OH_NN_INT32 : OH_NN_INT8; in ConvertType()
145 nnType = (nnTypeEquivalent == kTfLiteInt32) ? OH_NN_INT32 : OH_NN_UINT8; in ConvertType()
H A Dnnrt_op_builder.cpp193 OH_NN_DataType nnType {OH_NN_FLOAT32}; in AddQuantizeParams() local
196 m_pTensorMapping->ConvertType(m_context, inputIndex, 0, nnType); in AddQuantizeParams()
197 …mappingArgs.builder->AddScalarInt64Tensor(static_cast<int64_t>(nnType), OH_NN_QUANT_DTYPE_CAST_SRC… in AddQuantizeParams()
200 m_pTensorMapping->ConvertType(m_context, outputIndex, 0, nnType); in AddQuantizeParams()
201 …mappingArgs.builder->AddScalarInt64Tensor(static_cast<int64_t>(nnType), OH_NN_QUANT_DTYPE_CAST_DST… in AddQuantizeParams()
432 OH_NN_DataType nnType {OH_NN_UNKNOWN}; in ConstructNNTensor() local
433 … TF_LITE_ENSURE_STATUS(m_pTensorMapping->ConvertType(m_context, tensorIndex, tensorFlags, nnType)); in ConstructNNTensor()
456 nnTensor.dataType = nnType; in ConstructNNTensor()
H A Dnnrt_op_builder.h125 TfLiteStatus AddScalarTensor(T value, OH_NN_DataType nnType, OH_NN_TensorType nnTensorType) in AddScalarTensor() argument
128 .dataType = nnType, in AddScalarTensor()
148 TfLiteStatus AddVectorTensor(const T* values, int32_t numValues, OH_NN_DataType nnType, in AddVectorTensor() argument
167 .dataType = nnType, in AddVectorTensor()
H A Dnnrt_delegate_kernel.cpp296 OH_NN_DataType nnType {OH_NN_UNKNOWN}; in ConvertTensorTypeToNn() local
297 TF_LITE_ENSURE_STATUS(m_tensorMapping.ConvertType(context, indexPair.first, 0, nnType)); in ConvertTensorTypeToNn()
313 nnTensor.dataType = nnType; in ConvertTensorTypeToNn()