| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/cmsis_nn/ |
| A D | svdf.cc | 316 TF_LITE_ENSURE_EQ(context, num_filters % rank, 0); in Prepare()
323 TF_LITE_ENSURE_EQ(context, NumDimensions(input), 2); in Prepare()
327 TF_LITE_ENSURE_EQ(context, node->outputs->size, 1); in Prepare()
329 TF_LITE_ENSURE_EQ(context, NumDimensions(output), 2); in Prepare()
331 TF_LITE_ENSURE_EQ(context, output->dims->data[1], num_units); in Prepare()
338 TF_LITE_ENSURE_EQ(context, NumDimensions(weights_time), 2); in Prepare()
344 TF_LITE_ENSURE_EQ(context, bias->dims->data[0], num_units); in Prepare()
350 TF_LITE_ENSURE_EQ(context, activation_state->dims->data[1], in Prepare()
355 TF_LITE_ENSURE_EQ(context, node->inputs->size, 5); in Prepare()
365 TF_LITE_ENSURE_EQ(context, bias->type, kTfLiteInt32); in Prepare()
[all …]
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/ |
| A D | svdf_common.cc | 361 TF_LITE_ENSURE_EQ(context, num_filters % rank, 0); in PrepareSvdf()
368 TF_LITE_ENSURE_EQ(context, NumDimensions(input), 2); in PrepareSvdf()
372 TF_LITE_ENSURE_EQ(context, node->outputs->size, 1); in PrepareSvdf()
375 TF_LITE_ENSURE_EQ(context, NumDimensions(output), 2); in PrepareSvdf()
377 TF_LITE_ENSURE_EQ(context, output->dims->data[1], num_units); in PrepareSvdf()
384 TF_LITE_ENSURE_EQ(context, NumDimensions(weights_time), 2); in PrepareSvdf()
390 TF_LITE_ENSURE_EQ(context, bias->dims->data[0], num_units); in PrepareSvdf()
396 TF_LITE_ENSURE_EQ(context, activation_state->dims->data[1], in PrepareSvdf()
401 TF_LITE_ENSURE_EQ(context, node->inputs->size, 5); in PrepareSvdf()
411 TF_LITE_ENSURE_EQ(context, bias->type, kTfLiteInt32); in PrepareSvdf()
[all …]
|
| A D | ceil.cc | 36 TF_LITE_ENSURE_EQ(context, NumInputs(node), 1); in Prepare()
37 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
40 TF_LITE_ENSURE_EQ(context, output->bytes, input->bytes); in Prepare()
41 TF_LITE_ENSURE_EQ(context, output->dims->size, input->dims->size); in Prepare()
43 TF_LITE_ENSURE_EQ(context, output->dims->data[i], input->dims->data[i]); in Prepare()
|
| A D | round.cc | 36 TF_LITE_ENSURE_EQ(context, NumInputs(node), 1); in Prepare()
37 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
40 TF_LITE_ENSURE_EQ(context, output->bytes, input->bytes); in Prepare()
41 TF_LITE_ENSURE_EQ(context, output->dims->size, input->dims->size); in Prepare()
43 TF_LITE_ENSURE_EQ(context, output->dims->data[i], input->dims->data[i]); in Prepare()
|
| A D | softmax_common.cc | 37 TF_LITE_ENSURE_EQ(context, output->params.zero_point, 0); in CalculateSoftmaxParams()
43 TF_LITE_ENSURE_EQ(context, output->params.zero_point, -32768); in CalculateSoftmaxParams()
48 TF_LITE_ENSURE_EQ(context, output->params.zero_point, -128); in CalculateSoftmaxParams()
93 TF_LITE_ENSURE_EQ(context, NumInputs(node), 1); in SoftmaxPrepare()
94 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in SoftmaxPrepare()
120 TF_LITE_ENSURE_EQ(context, input->type, output->type); in SoftmaxPrepare()
125 TF_LITE_ENSURE_EQ(context, output->params.zero_point, 0); in SoftmaxPrepare()
|
| A D | exp.cc | 30 TF_LITE_ENSURE_EQ(context, NumInputs(node), 1); in Prepare()
31 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
38 TF_LITE_ENSURE_EQ(context, output->bytes, input->bytes); in Prepare()
39 TF_LITE_ENSURE_EQ(context, output->dims->size, input->dims->size); in Prepare()
41 TF_LITE_ENSURE_EQ(context, output->dims->data[i], input->dims->data[i]); in Prepare()
|
| A D | resize_nearest_neighbor.cc | 35 TF_LITE_ENSURE_EQ(context, NumInputs(node), 2); in Prepare()
36 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
44 TF_LITE_ENSURE_EQ(context, NumDimensions(input), 4); in Prepare()
45 TF_LITE_ENSURE_EQ(context, NumDimensions(size), 1); in Prepare()
46 TF_LITE_ENSURE_EQ(context, size->type, kTfLiteInt32); in Prepare()
47 TF_LITE_ENSURE_EQ(context, size->dims->data[0], 2); in Prepare()
|
| A D | depth_to_space.cc | 40 TF_LITE_ENSURE_EQ(context, NumInputs(node), 1); in CalculateOpData()
41 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in CalculateOpData()
49 TF_LITE_ENSURE_EQ(context, NumDimensions(input), 4); in CalculateOpData()
65 TF_LITE_ENSURE_EQ(context, input_height, output_height / block_size); in CalculateOpData()
66 TF_LITE_ENSURE_EQ(context, input_width, output_width / block_size); in CalculateOpData()
67 TF_LITE_ENSURE_EQ(context, input_channels, in CalculateOpData()
|
| A D | resize_bilinear.cc | 33 TF_LITE_ENSURE_EQ(context, NumInputs(node), 2); in Prepare()
34 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
40 TF_LITE_ENSURE_EQ(context, NumDimensions(input), 4); in Prepare()
41 TF_LITE_ENSURE_EQ(context, NumDimensions(size), 1); in Prepare()
43 TF_LITE_ENSURE_EQ(context, size->type, kTfLiteInt32); in Prepare()
|
| A D | fill.cc | 33 TF_LITE_ENSURE_EQ(context, array->data[i], GetTensorData<T>(tensor)[i]); in EnsureEqImpl()
42 TF_LITE_ENSURE_EQ(context, NumDimensions(tensor), 1); in EnsureEq()
44 TF_LITE_ENSURE_EQ(context, array->size, tensor_len); in EnsureEq()
80 TF_LITE_ENSURE_EQ(context, NumDimensions(value), 0); in Prepare()
83 TF_LITE_ENSURE_EQ(context, value->type, output->type); in Prepare()
|
| A D | reshape.cc | 58 TF_LITE_ENSURE_EQ(context, stretch_dim, -1); in ReshapeOutput()
70 TF_LITE_ENSURE_EQ(context, num_input_elements, num_output_elements); in ReshapeOutput()
76 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
77 TF_LITE_ENSURE_EQ(context, ReshapeOutput(context, node), kTfLiteOk); in Prepare()
|
| A D | pad.cc | 50 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
61 TF_LITE_ENSURE_EQ(context, input->type, output->type); in Prepare()
68 TF_LITE_ENSURE_EQ(context, input->type, constant_values->type); in Prepare()
70 TF_LITE_ENSURE_EQ(context, NumElements(constant_values), 1); in Prepare()
74 TF_LITE_ENSURE_EQ(context, GetTensorShape(paddings).FlatSize(), in Prepare()
86 TF_LITE_ENSURE_EQ(context, output_dim, expected_dim); in Prepare()
124 TF_LITE_ENSURE_EQ(context, output->params.zero_point, in Prepare()
126 TF_LITE_ENSURE_EQ(context, static_cast<double>(output->params.scale), in Prepare()
|
| A D | cumsum.cc | 47 TF_LITE_ENSURE_EQ(context, NumInputs(node), 2); in CalculateOpData()
48 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in CalculateOpData()
55 TF_LITE_ENSURE_EQ(context, axis->type, kTfLiteInt32); in CalculateOpData()
57 TF_LITE_ENSURE_EQ(context, NumElements(axis), 1); in CalculateOpData()
63 TF_LITE_ENSURE_EQ(context, input->type, output->type); in CalculateOpData()
|
| A D | circular_buffer.cc | 97 TF_LITE_ENSURE_EQ(context, input->dims->data[0], output->dims->data[0]); in Prepare()
98 TF_LITE_ENSURE_EQ(context, 1, input->dims->data[1]); in Prepare()
99 TF_LITE_ENSURE_EQ(context, input->dims->data[2], output->dims->data[2]); in Prepare()
100 TF_LITE_ENSURE_EQ(context, output->dims->data[3], input->dims->data[3]); in Prepare()
|
| A D | l2_pool_2d.cc | 41 TF_LITE_ENSURE_EQ(context, NumInputs(node), 1); in L2Prepare()
42 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in L2Prepare()
48 TF_LITE_ENSURE_EQ(context, NumDimensions(input), kTensorShapeRank); in L2Prepare()
49 TF_LITE_ENSURE_EQ(context, NumDimensions(output), kTensorShapeRank); in L2Prepare()
|
| A D | squeeze.cc | 40 TF_LITE_ENSURE_EQ(context, NumInputs(node), 1); in Prepare()
41 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
92 TF_LITE_ENSURE_EQ(context, op_context.input->bytes, op_context.output->bytes); in Eval()
|
| A D | shape.cc | 38 TF_LITE_ENSURE_EQ(context, NumInputs(node), 1); in Prepare()
39 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
|
| A D | strided_slice.cc | 112 TF_LITE_ENSURE_EQ(context, output_shape->data[shape_size], dim_shape); in CheckOutputSize()
116 TF_LITE_ENSURE_EQ(context, output_shape->size, shape_size); in CheckOutputSize()
129 TF_LITE_ENSURE_EQ(context, NumInputs(node), 4); in Prepare()
130 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
|
| A D | l2norm.cc | 49 TF_LITE_ENSURE_EQ(context, NumInputs(node), 1); in Prepare()
50 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
71 TF_LITE_ENSURE_EQ(context, params->activation, kTfLiteActNone); in Prepare()
|
| A D | zeros_like.cc | 28 TF_LITE_ENSURE_EQ(context, NumInputs(node), 1); in Prepare()
29 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
|
| A D | cast.cc | 28 TF_LITE_ENSURE_EQ(context, NumInputs(node), 1); in Prepare()
29 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
|
| A D | detection_postprocess.cc | 158 TF_LITE_ENSURE_EQ(context, NumInputs(node), 3); in Prepare()
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()
169 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 4); in Prepare()
220 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 4); in Prepare()
274 TF_LITE_ENSURE_EQ(context, input_box_encodings->dims->data[0], kBatchSize); in DecodeCenterSizeBoxes()
730 TF_LITE_ENSURE_EQ(context, input_class_predictions->dims->data[0], in NonMaxSuppressionMultiClass()
732 TF_LITE_ENSURE_EQ(context, input_class_predictions->dims->data[1], num_boxes); in NonMaxSuppressionMultiClass()
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/experimental/kernels/ |
| A D | ctc_beam_search_decoder.cc | 61 TF_LITE_ENSURE_EQ(context, NumInputs(node), 2); in Prepare()
63 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 3 * top_paths + 1); in Prepare()
68 TF_LITE_ENSURE_EQ(context, NumDimensions(inputs), 3); in Prepare()
70 TF_LITE_ENSURE_EQ(context, inputs->type, kTfLiteFloat32); in Prepare()
76 TF_LITE_ENSURE_EQ(context, NumDimensions(sequence_length), 1); in Prepare()
77 TF_LITE_ENSURE_EQ(context, NumElements(sequence_length), batch_size); in Prepare()
79 TF_LITE_ENSURE_EQ(context, sequence_length->type, kTfLiteInt32); in Prepare()
129 TF_LITE_ENSURE_EQ(context, batch_s.size(), top_paths); in StoreAllDecodedSequences()
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/experimental/resource/ |
| A D | static_hashtable.h | 60 TF_LITE_ENSURE_EQ(context, keys->type, key_type_); in CheckKeyAndValueTypes()
61 TF_LITE_ENSURE_EQ(context, values->type, value_type_); in CheckKeyAndValueTypes()
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/experimental/microfrontend/ |
| A D | audio_microfrontend.cc | 91 TF_LITE_ENSURE_EQ(context, NumInputs(node), 1); in Prepare()
92 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
100 TF_LITE_ENSURE_EQ(context, NumDimensions(input), 1); in Prepare()
102 TF_LITE_ENSURE_EQ(context, input->type, kTfLiteInt16); in Prepare()
|