| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/ |
| A D | batch_to_space_nd.cc | 48 TF_LITE_ENSURE(context, NumDimensions(input) >= kInputOutputMinDimensionNum); in Prepare()
49 TF_LITE_ENSURE(context, NumDimensions(output) >= kInputOutputMinDimensionNum); in Prepare()
50 TF_LITE_ENSURE(context, NumDimensions(input) <= kInputOutputMaxDimensionNum); in Prepare()
51 TF_LITE_ENSURE(context, NumDimensions(output) <= kInputOutputMaxDimensionNum); in Prepare()
|
| A D | space_to_batch_nd.cc | 54 TF_LITE_ENSURE(context, NumDimensions(input) >= kInputOutputMinDimensionNum); in Prepare()
55 TF_LITE_ENSURE(context, NumDimensions(output) >= kInputOutputMinDimensionNum); in Prepare()
56 TF_LITE_ENSURE(context, NumDimensions(input) <= kInputOutputMaxDimensionNum); in Prepare()
57 TF_LITE_ENSURE(context, NumDimensions(output) <= kInputOutputMaxDimensionNum); in Prepare()
|
| A D | resize_bilinear.cc | 40 TF_LITE_ENSURE_EQ(context, NumDimensions(input), 4); in Prepare()
41 TF_LITE_ENSURE_EQ(context, NumDimensions(size), 1); in Prepare()
|
| A D | fill.cc | 42 TF_LITE_ENSURE_EQ(context, NumDimensions(tensor), 1); in EnsureEq()
80 TF_LITE_ENSURE_EQ(context, NumDimensions(value), 0); in Prepare()
|
| A D | resize_nearest_neighbor.cc | 44 TF_LITE_ENSURE_EQ(context, NumDimensions(input), 4); in Prepare()
45 TF_LITE_ENSURE_EQ(context, NumDimensions(size), 1); in Prepare()
|
| A D | l2_pool_2d.cc | 48 TF_LITE_ENSURE_EQ(context, NumDimensions(input), kTensorShapeRank); in L2Prepare()
49 TF_LITE_ENSURE_EQ(context, NumDimensions(output), kTensorShapeRank); in L2Prepare()
|
| A D | svdf_common.cc | 368 TF_LITE_ENSURE_EQ(context, NumDimensions(input), 2); in PrepareSvdf()
375 TF_LITE_ENSURE_EQ(context, NumDimensions(output), 2); in PrepareSvdf()
380 TF_LITE_ENSURE_EQ(context, NumDimensions(weights_feature), 2); in PrepareSvdf()
384 TF_LITE_ENSURE_EQ(context, NumDimensions(weights_time), 2); in PrepareSvdf()
394 TF_LITE_ENSURE_EQ(context, NumDimensions(activation_state), 2); in PrepareSvdf()
|
| A D | gather_nd.cc | 63 const int params_rank = NumDimensions(params); in Prepare()
64 const int indices_rank = NumDimensions(indices); in Prepare()
|
| A D | concatenation.cc | 44 return NumDimensions(output_tensor) + axis; in CalculatePositiveAxis()
164 int num_dimensions = NumDimensions(input); in Prepare()
|
| A D | squeeze.cc | 44 const int input_num_dims = NumDimensions(op_context.input); in Prepare()
|
| A D | pad.cc | 64 TF_LITE_ENSURE(context, NumDimensions(input) <= in Prepare()
97 const int num_input_dimensions = NumDimensions(input); in Prepare()
|
| A D | depth_to_space.cc | 49 TF_LITE_ENSURE_EQ(context, NumDimensions(input), 4); in CalculateOpData()
|
| A D | l2norm.cc | 57 TF_LITE_ENSURE(context, NumDimensions(input) <= 4); in Prepare()
|
| A D | softmax_common.cc | 97 TF_LITE_ENSURE(context, NumDimensions(input) >= 1); in SoftmaxPrepare()
|
| A D | cumsum.cc | 59 TF_LITE_ENSURE(context, NumDimensions(input) >= 1); in CalculateOpData()
|
| A D | detection_postprocess.cc | 165 TF_LITE_ENSURE_EQ(context, NumDimensions(input_box_encodings), 3); in Prepare()
166 TF_LITE_ENSURE_EQ(context, NumDimensions(input_class_predictions), 3); in Prepare()
167 TF_LITE_ENSURE_EQ(context, NumDimensions(input_anchors), 2); in Prepare()
|
| A D | strided_slice.cc | 46 dims = NumDimensions(input); in StridedSliceContext()
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/cmsis_nn/ |
| A D | svdf.cc | 323 TF_LITE_ENSURE_EQ(context, NumDimensions(input), 2); in Prepare()
329 TF_LITE_ENSURE_EQ(context, NumDimensions(output), 2); in Prepare()
334 TF_LITE_ENSURE_EQ(context, NumDimensions(weights_feature), 2); in Prepare()
338 TF_LITE_ENSURE_EQ(context, NumDimensions(weights_time), 2); in Prepare()
348 TF_LITE_ENSURE_EQ(context, NumDimensions(activation_state), 2); in Prepare()
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/ |
| A D | kernel_util.cc | 433 int dims1 = NumDimensions(input1); in CalculateShapeForBroadcast()
434 int dims2 = NumDimensions(input2); in CalculateShapeForBroadcast()
463 int dims1 = NumDimensions(input1); in CalculateShapeForBroadcast()
464 int dims2 = NumDimensions(input2); in CalculateShapeForBroadcast()
465 int dims3 = NumDimensions(input3); in CalculateShapeForBroadcast()
|
| A D | kernel_util.h | 147 inline int NumDimensions(const TfLiteTensor* t) { return t->dims->size; } in NumDimensions() function
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/experimental/kernels/ |
| A D | ctc_beam_search_decoder.cc | 68 TF_LITE_ENSURE_EQ(context, NumDimensions(inputs), 3); in Prepare()
76 TF_LITE_ENSURE_EQ(context, NumDimensions(sequence_length), 1); in Prepare()
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/experimental/microfrontend/ |
| A D | audio_microfrontend.cc | 100 TF_LITE_ENSURE_EQ(context, NumDimensions(input), 1); in Prepare()
|