Home
last modified time | relevance | path

Searched refs:scalarAsTensor (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 Dtensor_mapping.h127 const bool scalarAsTensor = tensorFlags & NN_TENSOR_FLAG_SCALAR_AS_TENSOR; in ConvertType() local
134 if (scalarAsTensor) { in ConvertType()
H A Dnnrt_op_builder.cpp353 const bool scalarAsTensor = tensorFlags & NN_TENSOR_FLAG_SCALAR_AS_TENSOR; in AddTensor() local
359 …TF_LITE_ENSURE_STATUS(ConstructNNTensor(tensorIndex, builtinCode, scalarAsTensor, nnQuantParam, nn… in AddTensor()
445 const bool scalarAsTensor = tensorFlags & NN_TENSOR_FLAG_SCALAR_AS_TENSOR; in ConstructNNTensor() local
446 if (scalarAsTensor && tensorRank == 0) { in ConstructNNTensor()
H A Dnnrt_delegate_kernel.cpp258 const bool scalarAsTensor = IsScalarInputSupported(reg->builtin_code); in AddOpsAndTensors() local
260 if (scalarAsTensor) { in AddOpsAndTensors()