Searched refs:TransposeTensor (Results 1 – 2 of 2) sorted by relevance
| /ohos5.0/foundation/ai/neural_network_runtime/example/deep_learning_framework/tflite/delegates/nnrt_delegate/ |
| H A D | nnrt_op_builder.cpp | 411 … TransposeTensor(m_context, tensorIndex, tensorDims, reinterpret_cast<float*>(tensorData.data()))); in TransposeDepthwiseTensor()
414 …TransposeTensor(m_context, tensorIndex, tensorDims, reinterpret_cast<int32_t*>(tensorData.data()))… in TransposeDepthwiseTensor()
417 …TransposeTensor(m_context, tensorIndex, tensorDims, reinterpret_cast<int8_t*>(tensorData.data()))); in TransposeDepthwiseTensor()
420 …TransposeTensor(m_context, tensorIndex, tensorDims, reinterpret_cast<uint8_t*>(tensorData.data()))… in TransposeDepthwiseTensor()
|
| H A D | nnrt_utils.h | 110 TfLiteStatus TransposeTensor(TfLiteContext* context, int32_t tensorIndex, const int32_t* dims, in TransposeTensor() function
|