Searched defs:tensorIndex (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 D | tensor_mapping.h | 93 …TfLiteStatus ConvertQuantParams(TfLiteContext* context, int32_t tensorIndex, OH_NN_QuantParam& qua… in ConvertQuantParams() 125 …TfLiteStatus ConvertType(TfLiteContext* context, int32_t tensorIndex, int32_t tensorFlags, OH_NN_D… in ConvertType()
|
H A D | nnrt_op_builder.cpp | 326 TfLiteStatus NnrtOpBuilder::AddTensor(int32_t tensorIndex, int32_t builtinCode, std::vector<uint32_… in AddTensor() 349 TfLiteStatus NnrtOpBuilder::AddTensor(int32_t tensorIndex, int32_t builtinCode, int32_t tensorFlags, in AddTensor() 398 TfLiteStatus NnrtOpBuilder::TransposeDepthwiseTensor(int32_t tensorIndex, OH_NN_Tensor& nnTensor, in TransposeDepthwiseTensor() 429 TfLiteStatus NnrtOpBuilder::ConstructNNTensor(int32_t tensorIndex, int32_t builtinCode, int32_t ten… in ConstructNNTensor()
|
H A D | nnrt_utils.h | 110 TfLiteStatus TransposeTensor(TfLiteContext* context, int32_t tensorIndex, const int32_t* dims, in TransposeTensor()
|
/ohos5.0/foundation/ai/neural_network_runtime/example/deep_learning_framework/tflite/tools/ |
H A D | utils.cpp | 256 bool IsEqualShape(int tensorIndex, const std::vector<int>& dims, std::unique_ptr<tflite::Interprete… in IsEqualShape()
|