Searched refs:SizeOfDimension (Results 1 – 7 of 7) sorted by relevance
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/ |
| A D | l2_pool_2d.cc | 52 int batches = SizeOfDimension(input, kBatchRank); in L2Prepare()
53 int height = SizeOfDimension(input, kHeightRank); in L2Prepare()
54 int width = SizeOfDimension(input, kWidthRank); in L2Prepare()
55 int channels_out = SizeOfDimension(input, kChannelRank); in L2Prepare()
|
| A D | gather_nd.cc | 65 const int indices_nd = SizeOfDimension(indices, indices_rank - 1); in Prepare()
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/cmsis_nn/ |
| A D | depthwise_conv.cc | 65 int input_width = SizeOfDimension(input, 2); in Prepare()
66 int input_height = SizeOfDimension(input, 1); in Prepare()
67 int filter_width = SizeOfDimension(filter, 2); in Prepare()
68 int filter_height = SizeOfDimension(filter, 1); in Prepare()
69 int output_width = SizeOfDimension(output, 2); in Prepare()
70 int output_height = SizeOfDimension(output, 1); in Prepare()
|
| A D | pooling.cc | 52 int height = SizeOfDimension(input, 1); in CalculateOpData()
53 int width = SizeOfDimension(input, 2); in CalculateOpData()
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/experimental/kernels/ |
| A D | ctc_beam_search_decoder.cc | 71 const int batch_size = SizeOfDimension(inputs, 1); in Prepare()
187 const int max_time = SizeOfDimension(inputs, 0); in Eval()
188 const int batch_size = SizeOfDimension(inputs, 1); in Eval()
189 const int num_classes = SizeOfDimension(inputs, 2); in Eval()
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/ |
| A D | kernel_util.cc | 443 int d1 = i >= dims1 ? 1 : SizeOfDimension(input1, dims1 - i - 1); in CalculateShapeForBroadcast()
444 int d2 = i >= dims2 ? 1 : SizeOfDimension(input2, dims2 - i - 1); in CalculateShapeForBroadcast()
470 int d1 = i >= dims1 ? 1 : SizeOfDimension(input1, dims1 - i - 1); in CalculateShapeForBroadcast()
471 int d2 = i >= dims2 ? 1 : SizeOfDimension(input2, dims2 - i - 1); in CalculateShapeForBroadcast()
472 int d3 = i >= dims3 ? 1 : SizeOfDimension(input3, dims3 - i - 1); in CalculateShapeForBroadcast()
|
| A D | kernel_util.h | 148 inline int SizeOfDimension(const TfLiteTensor* t, int dim) { in SizeOfDimension() function
|
Completed in 9 milliseconds