Home
last modified time | relevance | path

Searched refs:nnTensorType (Results 1 – 3 of 3) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/example/deep_learning_framework/tflite/delegates/nnrt_delegate/
H A Dnnrt_op_builder.h65 return AddScalarTensor<bool>(value, OH_NN_BOOL, nnTensorType); in AddScalarBoolTensor()
69 return AddScalarTensor<int32_t>(value, OH_NN_INT32, nnTensorType); in AddScalarInt32Tensor()
73 return AddScalarTensor<int8_t>(value, OH_NN_INT8, nnTensorType); in AddScalarInt8Tensor()
77 return AddScalarTensor<int64_t>(value, OH_NN_INT64, nnTensorType); in AddScalarInt64Tensor()
81 return AddScalarTensor<float>(value, OH_NN_FLOAT32, nnTensorType); in AddScalarFloat32Tensor()
132 .type = nnTensorType, in AddScalarTensor()
149 OH_NN_TensorType nnTensorType) in AddVectorTensor() argument
171 .type = nnTensorType, in AddVectorTensor()
187 int32_t builtinCode, OH_NN_TensorType nnTensorType) in AddActivateParamsInOperator() argument
211 OH_NN_TensorType nnTensorType) in AddPadParamsInOperator() argument
[all …]
/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_core/
H A Dvalidation.cpp61 bool ValidateTensorType(OH_NN_TensorType nnTensorType) in ValidateTensorType() argument
63 if ((nnTensorType >= OH_NN_TENSOR) && (nnTensorType <= OH_NN_REDUCE_L2_COEFF)) { in ValidateTensorType()
H A Dvalidation.h36 bool ValidateTensorType(OH_NN_TensorType nnTensorType);