Home
last modified time | relevance | path

Searched refs:TF_LITE_ENSURE_EQ (Results 1 – 25 of 186) sorted by relevance

12345678

/external/tensorflow/tensorflow/lite/micro/kernels/
Dsvdf.cc225 TF_LITE_ENSURE_EQ(context, num_filters % rank, 0); in Prepare()
232 TF_LITE_ENSURE_EQ(context, NumDimensions(input), 2); in Prepare()
236 TF_LITE_ENSURE_EQ(context, node->outputs->size, 1); in Prepare()
239 TF_LITE_ENSURE_EQ(context, NumDimensions(output), 2); in Prepare()
240 TF_LITE_ENSURE_EQ(context, output->dims->data[0], batch_size); in Prepare()
241 TF_LITE_ENSURE_EQ(context, output->dims->data[1], num_units); in Prepare()
244 TF_LITE_ENSURE_EQ(context, NumDimensions(weights_feature), 2); in Prepare()
245 TF_LITE_ENSURE_EQ(context, weights_feature->dims->data[1], input_size); in Prepare()
248 TF_LITE_ENSURE_EQ(context, NumDimensions(weights_time), 2); in Prepare()
249 TF_LITE_ENSURE_EQ(context, weights_time->dims->data[0], num_filters); in Prepare()
[all …]
Dsoftmax_common.cc39 TF_LITE_ENSURE_EQ(context, output->params.zero_point, 0); in CalculateSoftmaxParams()
41 TF_LITE_ENSURE_EQ(context, output->params.zero_point, 0); in CalculateSoftmaxParams()
47 TF_LITE_ENSURE_EQ(context, output->params.zero_point, -32768); in CalculateSoftmaxParams()
52 TF_LITE_ENSURE_EQ(context, output->params.zero_point, -128); in CalculateSoftmaxParams()
97 TF_LITE_ENSURE_EQ(context, NumInputs(node), 1); in SoftmaxPrepare()
98 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in SoftmaxPrepare()
125 TF_LITE_ENSURE_EQ(context, input->type, output->type); in SoftmaxPrepare()
130 TF_LITE_ENSURE_EQ(context, output->params.zero_point, 0); in SoftmaxPrepare()
Dceil.cc36 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()
Dround.cc36 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()
Dresize_nearest_neighbor.cc35 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()
Dexp.cc30 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()
/external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/
Dsvdf.cc312 TF_LITE_ENSURE_EQ(context, num_filters % rank, 0); in Prepare()
319 TF_LITE_ENSURE_EQ(context, NumDimensions(input), 2); in Prepare()
323 TF_LITE_ENSURE_EQ(context, node->outputs->size, 1); in Prepare()
325 TF_LITE_ENSURE_EQ(context, NumDimensions(output), 2); in Prepare()
326 TF_LITE_ENSURE_EQ(context, output->dims->data[0], batch_size); in Prepare()
327 TF_LITE_ENSURE_EQ(context, output->dims->data[1], num_units); in Prepare()
330 TF_LITE_ENSURE_EQ(context, NumDimensions(weights_feature), 2); in Prepare()
331 TF_LITE_ENSURE_EQ(context, weights_feature->dims->data[1], input_size); in Prepare()
334 TF_LITE_ENSURE_EQ(context, NumDimensions(weights_time), 2); in Prepare()
335 TF_LITE_ENSURE_EQ(context, weights_time->dims->data[0], num_filters); in Prepare()
[all …]
/external/tensorflow/tensorflow/lite/kernels/
Dembedding_lookup_sparse.cc83 TF_LITE_ENSURE_EQ(context, NumInputs(node), 5); in Prepare()
84 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
88 TF_LITE_ENSURE_EQ(context, NumDimensions(ids), 1); in Prepare()
89 TF_LITE_ENSURE_EQ(context, ids->type, kTfLiteInt32); in Prepare()
93 TF_LITE_ENSURE_EQ(context, NumDimensions(indices), 2); in Prepare()
94 TF_LITE_ENSURE_EQ(context, indices->type, kTfLiteInt32); in Prepare()
98 TF_LITE_ENSURE_EQ(context, NumDimensions(shape), 1); in Prepare()
99 TF_LITE_ENSURE_EQ(context, shape->type, kTfLiteInt32); in Prepare()
103 TF_LITE_ENSURE_EQ(context, NumDimensions(weights), 1); in Prepare()
104 TF_LITE_ENSURE_EQ(context, weights->type, kTfLiteFloat32); in Prepare()
[all …]
Dnon_max_suppression.cc85 TF_LITE_ENSURE_EQ(context, input_boxes->type, kTfLiteFloat32); in Prepare()
86 TF_LITE_ENSURE_EQ(context, NumDimensions(input_boxes), 2); in Prepare()
87 TF_LITE_ENSURE_EQ(context, SizeOfDimension(input_boxes, 1), 4); in Prepare()
92 TF_LITE_ENSURE_EQ(context, input_scores->type, kTfLiteFloat32); in Prepare()
93 TF_LITE_ENSURE_EQ(context, NumDimensions(input_scores), 1); in Prepare()
94 TF_LITE_ENSURE_EQ(context, num_boxes, SizeOfDimension(input_scores, 0)); in Prepare()
101 TF_LITE_ENSURE_EQ(context, input_max_output_size->type, kTfLiteInt32); in Prepare()
102 TF_LITE_ENSURE_EQ(context, NumDimensions(input_max_output_size), 0); in Prepare()
115 TF_LITE_ENSURE_EQ(context, input_iou_threshold->type, kTfLiteFloat32); in Prepare()
116 TF_LITE_ENSURE_EQ(context, NumDimensions(input_iou_threshold), 0); in Prepare()
[all …]
Dbidirectional_sequence_lstm.cc200 TF_LITE_ENSURE_EQ(context, input_to_forget_weights->dims->size, 2); in CheckLstmTensorDimensionsAndTypes()
201 TF_LITE_ENSURE_EQ(context, input_to_forget_weights->dims->data[0], n_cell); in CheckLstmTensorDimensionsAndTypes()
202 TF_LITE_ENSURE_EQ(context, input_to_forget_weights->dims->data[1], n_input); in CheckLstmTensorDimensionsAndTypes()
210 TF_LITE_ENSURE_EQ(context, input_to_input_weights->dims->size, 2); in CheckLstmTensorDimensionsAndTypes()
211 TF_LITE_ENSURE_EQ(context, input_to_input_weights->dims->data[0], n_cell); in CheckLstmTensorDimensionsAndTypes()
212 TF_LITE_ENSURE_EQ(context, input_to_input_weights->dims->data[1], n_input); in CheckLstmTensorDimensionsAndTypes()
221 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->size, 2); in CheckLstmTensorDimensionsAndTypes()
222 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->data[0], n_cell); in CheckLstmTensorDimensionsAndTypes()
223 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->data[1], n_input); in CheckLstmTensorDimensionsAndTypes()
231 TF_LITE_ENSURE_EQ(context, input_to_output_weights->dims->size, 2); in CheckLstmTensorDimensionsAndTypes()
[all …]
Dhashtable_lookup.cc55 TF_LITE_ENSURE_EQ(context, NumInputs(node), 3); in Prepare()
56 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 2); in Prepare()
60 TF_LITE_ENSURE_EQ(context, NumDimensions(lookup), 1); in Prepare()
61 TF_LITE_ENSURE_EQ(context, lookup->type, kTfLiteInt32); in Prepare()
65 TF_LITE_ENSURE_EQ(context, NumDimensions(key), 1); in Prepare()
66 TF_LITE_ENSURE_EQ(context, key->type, kTfLiteInt32); in Prepare()
71 TF_LITE_ENSURE_EQ(context, SizeOfDimension(key, 0), in Prepare()
74 TF_LITE_ENSURE_EQ(context, NumDimensions(value), 1); in Prepare()
79 TF_LITE_ENSURE_EQ(context, hits->type, kTfLiteUInt8); in Prepare()
85 TF_LITE_ENSURE_EQ(context, value->type, output->type); in Prepare()
Dbatch_to_space_nd.cc64 TF_LITE_ENSURE_EQ(context, NumDimensions(op_context->block_shape), 1); in ResizeOutputTensor()
65 TF_LITE_ENSURE_EQ(context, op_context->block_shape->dims->data[0], in ResizeOutputTensor()
68 TF_LITE_ENSURE_EQ(context, NumDimensions(op_context->crops), 2); in ResizeOutputTensor()
69 TF_LITE_ENSURE_EQ(context, op_context->crops->dims->data[0], in ResizeOutputTensor()
71 TF_LITE_ENSURE_EQ(context, op_context->crops->dims->data[1], 2); in ResizeOutputTensor()
81 TF_LITE_ENSURE_EQ(context, output_batch_size % block_shape[dim], 0); in ResizeOutputTensor()
95 TF_LITE_ENSURE_EQ(context, NumInputs(node), 3); in Prepare()
96 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
103 TF_LITE_ENSURE_EQ(context, op_context.input->type, op_context.output->type); in Prepare()
Dunidirectional_sequence_lstm.cc424 TF_LITE_ENSURE_EQ(context, input_to_input_weights->dims->size, 2); in CheckInputTensorDimensions()
425 TF_LITE_ENSURE_EQ(context, input_to_input_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
426 TF_LITE_ENSURE_EQ(context, input_to_input_weights->dims->data[1], n_input); in CheckInputTensorDimensions()
434 TF_LITE_ENSURE_EQ(context, input_to_forget_weights->dims->size, 2); in CheckInputTensorDimensions()
435 TF_LITE_ENSURE_EQ(context, input_to_forget_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
436 TF_LITE_ENSURE_EQ(context, input_to_forget_weights->dims->data[1], n_input); in CheckInputTensorDimensions()
442 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->size, 2); in CheckInputTensorDimensions()
443 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
444 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->data[1], n_input); in CheckInputTensorDimensions()
449 TF_LITE_ENSURE_EQ(context, recurrent_to_input_weights->dims->size, 2); in CheckInputTensorDimensions()
[all …]
Dspace_to_batch_nd.cc65 TF_LITE_ENSURE_EQ(context, NumDimensions(op_context->block_shape), 1); in ResizeOutputTensor()
66 TF_LITE_ENSURE_EQ(context, op_context->block_shape->dims->data[0], in ResizeOutputTensor()
69 TF_LITE_ENSURE_EQ(context, NumDimensions(op_context->paddings), 2); in ResizeOutputTensor()
70 TF_LITE_ENSURE_EQ(context, op_context->paddings->dims->data[0], in ResizeOutputTensor()
72 TF_LITE_ENSURE_EQ(context, op_context->paddings->dims->data[1], 2); in ResizeOutputTensor()
82 TF_LITE_ENSURE_EQ(context, final_dim_size % block_shape[dim], 0); in ResizeOutputTensor()
95 TF_LITE_ENSURE_EQ(context, NumInputs(node), 3); in Prepare()
96 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
Dwhile.cc50 TF_LITE_ENSURE_EQ(context, src_tensor_indices.size(), in CopyTensorsShapeAndType()
78 TF_LITE_ENSURE_EQ(context, src_tensor_indices.size(), in CopyTensorsData()
87 TF_LITE_ENSURE_EQ(context, src_tensor->bytes, dst_tensor->bytes); in CopyTensorsData()
102 TF_LITE_ENSURE_EQ(context, cond_output->dims->size, 1); in CheckCondOutput()
103 TF_LITE_ENSURE_EQ(context, cond_output->dims->data[0], 1); in CheckCondOutput()
134 TF_LITE_ENSURE_EQ(context, node->outputs->size, num_inputs); in Prepare()
146 TF_LITE_ENSURE_EQ(context, cond_subgraph->inputs().size(), num_inputs); in Prepare()
147 TF_LITE_ENSURE_EQ(context, cond_subgraph->outputs().size(), 1); in Prepare()
150 TF_LITE_ENSURE_EQ(context, body_subgraph->inputs().size(), num_inputs); in Prepare()
151 TF_LITE_ENSURE_EQ(context, body_subgraph->outputs().size(), num_inputs); in Prepare()
Dfill.cc72 TF_LITE_ENSURE_EQ(context, NumInputs(node), 2); in Prepare()
73 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
81 TF_LITE_ENSURE_EQ(context, NumDimensions(dims), 1); in Prepare()
88 TF_LITE_ENSURE_EQ(context, NumDimensions(value), 0); in Prepare()
95 TF_LITE_ENSURE_EQ(context, output->params.scale, value->params.scale); in Prepare()
96 TF_LITE_ENSURE_EQ(context, output->params.zero_point, in Prepare()
100 TF_LITE_ENSURE_EQ(context, value->params.zero_point, 0); in Prepare()
Dconcatenation.cc55 TF_LITE_ENSURE_EQ(context, params->activation, kTfLiteActNone); in Prepare()
68 TF_LITE_ENSURE_EQ(context, t->dims->size, t0->dims->size); in Prepare()
69 TF_LITE_ENSURE_EQ(context, t->type, input_type); in Prepare()
74 TF_LITE_ENSURE_EQ(context, t->dims->data[d], t0->dims->data[d]); in Prepare()
95 TF_LITE_ENSURE_EQ(context, t->params.scale, output->params.scale); in Prepare()
96 TF_LITE_ENSURE_EQ(context, t->params.zero_point, in Prepare()
105 TF_LITE_ENSURE_EQ(context, t->params.zero_point, 0); in Prepare()
107 TF_LITE_ENSURE_EQ(context, output->params.zero_point, 0); in Prepare()
Dlstm.cc819 TF_LITE_ENSURE_EQ(context, cell_state_params->scale->data[0], 1.0 / 32768); in PopulateQuantizedLstmParams8x8_8()
868 TF_LITE_ENSURE_EQ(context, input_to_forget_weights->dims->size, 2); in CheckInputTensorDimensions()
869 TF_LITE_ENSURE_EQ(context, input_to_forget_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
870 TF_LITE_ENSURE_EQ(context, input_to_forget_weights->dims->data[1], n_input); in CheckInputTensorDimensions()
879 TF_LITE_ENSURE_EQ(context, input_to_input_weights->dims->size, 2); in CheckInputTensorDimensions()
880 TF_LITE_ENSURE_EQ(context, input_to_input_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
881 TF_LITE_ENSURE_EQ(context, input_to_input_weights->dims->data[1], n_input); in CheckInputTensorDimensions()
890 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->size, 2); in CheckInputTensorDimensions()
891 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
892 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->data[1], n_input); in CheckInputTensorDimensions()
[all …]
Ddepth_to_space.cc45 TF_LITE_ENSURE_EQ(context, NumInputs(node), 1); in Prepare()
46 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
54 TF_LITE_ENSURE_EQ(context, NumDimensions(input), 4); in Prepare()
71 TF_LITE_ENSURE_EQ(context, input_height, output_height / block_size); in Prepare()
72 TF_LITE_ENSURE_EQ(context, input_width, output_width / block_size); in Prepare()
73 TF_LITE_ENSURE_EQ(context, input_channels, in Prepare()
Dl2norm.cc44 TF_LITE_ENSURE_EQ(context, NumInputs(node), 1); in Prepare()
45 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
61 TF_LITE_ENSURE_EQ(context, output->params.scale, (1. / 128.)); in Prepare()
63 TF_LITE_ENSURE_EQ(context, output->params.zero_point, 128); in Prepare()
66 TF_LITE_ENSURE_EQ(context, output->params.zero_point, 0); in Prepare()
72 TF_LITE_ENSURE_EQ(context, params->activation, kTfLiteActNone); in Prepare()
/external/tensorflow/tensorflow/lite/micro/kernels/xtensa/
Dsvdf.cc284 TF_LITE_ENSURE_EQ(context, in EvalIntegerSvdfHifi4()
305 TF_LITE_ENSURE_EQ( in EvalIntegerSvdfHifi4()
352 TF_LITE_ENSURE_EQ(context, input_size % 2, 0); in Prepare()
356 TF_LITE_ENSURE_EQ(context, num_filters % rank, 0); in Prepare()
368 TF_LITE_ENSURE_EQ(context, NumDimensions(input), 2); in Prepare()
372 TF_LITE_ENSURE_EQ(context, node->outputs->size, 1); in Prepare()
374 TF_LITE_ENSURE_EQ(context, NumDimensions(output), 2); in Prepare()
375 TF_LITE_ENSURE_EQ(context, output->dims->data[0], batch_size); in Prepare()
376 TF_LITE_ENSURE_EQ(context, output->dims->data[1], num_units); in Prepare()
379 TF_LITE_ENSURE_EQ(context, NumDimensions(weights_feature), 2); in Prepare()
[all …]
/external/tensorflow/tensorflow/lite/experimental/kernels/
Dunidirectional_sequence_gru.cc126 TF_LITE_ENSURE_EQ(context, node->inputs->size, kInputNum); in Prepare()
127 TF_LITE_ENSURE_EQ(context, node->outputs->size, kOutputNum); in Prepare()
132 TF_LITE_ENSURE_EQ(context, input->dims->size, 3); in Prepare()
141 TF_LITE_ENSURE_EQ(context, input_state->dims->size, 2); in Prepare()
142 TF_LITE_ENSURE_EQ(context, input_state->dims->data[0], n_batch); in Prepare()
149 TF_LITE_ENSURE_EQ(context, gate_weight->dims->size, 2); in Prepare()
150 TF_LITE_ENSURE_EQ(context, gate_weight->dims->data[0], 2 * n_output); in Prepare()
151 TF_LITE_ENSURE_EQ(context, gate_weight->dims->data[1], n_input + n_output); in Prepare()
157 TF_LITE_ENSURE_EQ(context, gate_bias->dims->size, 1); in Prepare()
158 TF_LITE_ENSURE_EQ(context, gate_bias->dims->data[0], 2 * n_output); in Prepare()
[all …]
/external/tensorflow/tensorflow/lite/kernels/perception/
Dmax_unpooling_2d.cc59 TF_LITE_ENSURE_EQ(context, NumInputs(node), 2); in Prepare()
60 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
67 TF_LITE_ENSURE_EQ(context, NumDimensions(indices), 4); in Prepare()
68 TF_LITE_ENSURE_EQ(context, NumDimensions(input), 4); in Prepare()
69 TF_LITE_ENSURE_EQ(context, input->type, kTfLiteFloat32); in Prepare()
70 TF_LITE_ENSURE_EQ(context, output->type, kTfLiteFloat32); in Prepare()
71 TF_LITE_ENSURE_EQ(context, indices->type, kTfLiteInt32); in Prepare()
Ddense_image_warp.cc90 TF_LITE_ENSURE_EQ(context, NumInputs(node), 2); in Prepare()
91 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
106 TF_LITE_ENSURE_EQ(context, NumDimensions(flow), 4); in Prepare()
107 TF_LITE_ENSURE_EQ(context, NumDimensions(input), 4); in Prepare()
110 TF_LITE_ENSURE_EQ(context, input_shape.Dims(0), flow_shape.Dims(0)); in Prepare()
111 TF_LITE_ENSURE_EQ(context, input_shape.Dims(1), flow_shape.Dims(1)); in Prepare()
112 TF_LITE_ENSURE_EQ(context, input_shape.Dims(2), flow_shape.Dims(2)); in Prepare()
/external/tensorflow/tensorflow/lite/kernels/hashtable/
Dhashtable_size.cc32 TF_LITE_ENSURE_EQ(context, NumInputs(node), 1); in PrepareHashtableSize()
33 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in PrepareHashtableSize()
40 TF_LITE_ENSURE_EQ(context, NumDimensions(input_resource_id_tensor), 1); in PrepareHashtableSize()
41 TF_LITE_ENSURE_EQ(context, SizeOfDimension(input_resource_id_tensor, 0), 1); in PrepareHashtableSize()
46 TF_LITE_ENSURE_EQ(context, output_tensor->type, kTfLiteInt64); in PrepareHashtableSize()

12345678