Home
last modified time | relevance | path

Searched refs:tensorIndex (Results 1 – 6 of 6) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/example/deep_learning_framework/tflite/delegates/nnrt_delegate/
H A Dnnrt_op_builder.cpp329 int32_t nnTensorIndex = m_pTensorMapping->LiteIndexToNn(tensorIndex); in AddTensor()
336 TfLiteTensor* tensor = &(m_context->tensors[tensorIndex]); in AddTensor()
342 TF_LITE_ENSURE_STATUS(AddTensor(tensorIndex, builtinCode, tensorFlags, nnTensorIndex)); in AddTensor()
349 TfLiteStatus NnrtOpBuilder::AddTensor(int32_t tensorIndex, int32_t builtinCode, int32_t tensorFlags, in AddTensor() argument
352 TfLiteTensor* tensor = &(m_context->tensors[tensorIndex]); in AddTensor()
385 nnTensorIndex = m_pTensorMapping->AddNewNnTensorIndex(tensorIndex); in AddTensor()
398 TfLiteStatus NnrtOpBuilder::TransposeDepthwiseTensor(int32_t tensorIndex, OH_NN_Tensor& nnTensor, in TransposeDepthwiseTensor() argument
408 TfLiteTensor* tensor = &(m_context->tensors[tensorIndex]); in TransposeDepthwiseTensor()
433 … TF_LITE_ENSURE_STATUS(m_pTensorMapping->ConvertType(m_context, tensorIndex, tensorFlags, nnType)); in ConstructNNTensor()
436 TfLiteTensor* tensor = &(m_context->tensors[tensorIndex]); in ConstructNNTensor()
[all …]
H A Dtensor_mapping.h93 …TfLiteStatus ConvertQuantParams(TfLiteContext* context, int32_t tensorIndex, OH_NN_QuantParam& qua… in ConvertQuantParams() argument
95 TfLiteTensor* tensor = &(context->tensors[tensorIndex]); in ConvertQuantParams()
125 …TfLiteStatus ConvertType(TfLiteContext* context, int32_t tensorIndex, int32_t tensorFlags, OH_NN_D… in ConvertType() argument
128 TfLiteTensor* tensor = &(context->tensors[tensorIndex]); in ConvertType()
129 TfLiteType nnTypeEquivalent = GetEqualLiteTypeFromLiteIndex(tensorIndex); in ConvertType()
136 AddTypeConversion(tensorIndex, kTfLiteFloat32); in ConvertType()
H A Dnnrt_op_builder.h99 TfLiteStatus AddTensorInput(int32_t tensorIndex, int32_t builtinCode, int32_t tensorFlags = 0)
101 return AddTensor(tensorIndex, builtinCode, m_augmentedInputs, tensorFlags);
104 TfLiteStatus AddTensorOutput(int32_t tensorIndex, int32_t builtinCode, int32_t tensorFlags = 0)
106 return AddTensor(tensorIndex, builtinCode, m_augmentedOutputs, tensorFlags);
273 TfLiteStatus AddTensor(int32_t tensorIndex, int32_t builtinCode, std::vector<uint32_t>& indices,
278 …TfLiteStatus AddTensor(int32_t tensorIndex, int32_t builtinCode, int32_t tensorFlags, int32_t& nnT…
281 …TfLiteStatus TransposeDepthwiseTensor(int32_t tensorIndex, OH_NN_Tensor& nnTensor, std::vector<int…
285 TfLiteStatus ConstructNNTensor(int32_t tensorIndex, int32_t builtinCode, int32_t tensorFlags,
H A Dnnrt_utils.h110 TfLiteStatus TransposeTensor(TfLiteContext* context, int32_t tensorIndex, const int32_t* dims, in TransposeTensor() argument
116 TfLiteTensor* tensor = &(context->tensors[tensorIndex]); in TransposeTensor()
/ohos5.0/foundation/ai/neural_network_runtime/example/deep_learning_framework/tflite/tools/
H A Dutils.h36 bool IsEqualShape(int tensorIndex, const std::vector<int>& dim, std::unique_ptr<tflite::Interpreter…
H A Dutils.cpp256 bool IsEqualShape(int tensorIndex, const std::vector<int>& dims, std::unique_ptr<tflite::Interprete… in IsEqualShape() argument
258 TfLiteTensor* tensor = interpreter->tensor(tensorIndex); in IsEqualShape()