Searched refs:dimCount (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_utils.cpp | 142 TfLiteStatus TransposeDims(TfLiteContext* context, const int32_t* dims, uint32_t dimCount, in TransposeDims() argument
147 if (dimCount != destAxis.size()) { in TransposeDims()
148 TFLITE_LOG_PROD(TFLITE_LOG_ERROR, "[NNRT-UTILS] Invalid dimension count %d.", dimCount); in TransposeDims()
163 TfLiteStatus GetTensorSize(TfLiteContext* context, const int32_t* dims, int32_t dimCount, int64_t& … in GetTensorSize() argument
167 if (dimCount != DEPTHWISE_WEIGHT_DIMENSION_COUNT) { in GetTensorSize()
174 for (int32_t i = 0; i < dimCount; ++i) { in GetTensorSize()
|
| H A D | nnrt_utils.h | 98 extern TfLiteStatus TransposeDims(TfLiteContext* context, const int32_t* dims, uint32_t dimCount,
104 extern TfLiteStatus GetTensorSize(TfLiteContext* context, const int32_t* dims, int32_t dimCount, in…
|