Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/contrib/lite/kernels/
Dunidirectional_sequence_lstm.cc88 TF_LITE_ENSURE_EQ(context, input_to_input_weights->dims->size, 2); in CheckInputTensorDimensions()
89 TF_LITE_ENSURE_EQ(context, input_to_input_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
90 TF_LITE_ENSURE_EQ(context, input_to_input_weights->dims->data[1], n_input); in CheckInputTensorDimensions()
95 TF_LITE_ENSURE_EQ(context, input_to_forget_weights->dims->size, 2); in CheckInputTensorDimensions()
96 TF_LITE_ENSURE_EQ(context, input_to_forget_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
97 TF_LITE_ENSURE_EQ(context, input_to_forget_weights->dims->data[1], n_input); in CheckInputTensorDimensions()
101 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->size, 2); in CheckInputTensorDimensions()
102 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
103 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->data[1], n_input); in CheckInputTensorDimensions()
108 TF_LITE_ENSURE_EQ(context, recurrent_to_input_weights->dims->size, 2); in CheckInputTensorDimensions()
[all …]
Dlstm.cc88 TF_LITE_ENSURE_EQ(context, input_to_input_weights->dims->size, 2); in CheckInputTensorDimensions()
89 TF_LITE_ENSURE_EQ(context, input_to_input_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
90 TF_LITE_ENSURE_EQ(context, input_to_input_weights->dims->data[1], n_input); in CheckInputTensorDimensions()
95 TF_LITE_ENSURE_EQ(context, input_to_forget_weights->dims->size, 2); in CheckInputTensorDimensions()
96 TF_LITE_ENSURE_EQ(context, input_to_forget_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
97 TF_LITE_ENSURE_EQ(context, input_to_forget_weights->dims->data[1], n_input); in CheckInputTensorDimensions()
101 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->size, 2); in CheckInputTensorDimensions()
102 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
103 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->data[1], n_input); in CheckInputTensorDimensions()
108 TF_LITE_ENSURE_EQ(context, recurrent_to_input_weights->dims->size, 2); in CheckInputTensorDimensions()
[all …]
Dembedding_lookup_sparse.cc81 TF_LITE_ENSURE_EQ(context, NumInputs(node), 5); in Prepare()
82 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
85 TF_LITE_ENSURE_EQ(context, NumDimensions(ids), 1); in Prepare()
86 TF_LITE_ENSURE_EQ(context, ids->type, kTfLiteInt32); in Prepare()
89 TF_LITE_ENSURE_EQ(context, NumDimensions(indices), 2); in Prepare()
90 TF_LITE_ENSURE_EQ(context, indices->type, kTfLiteInt32); in Prepare()
93 TF_LITE_ENSURE_EQ(context, NumDimensions(shape), 1); in Prepare()
94 TF_LITE_ENSURE_EQ(context, shape->type, kTfLiteInt32); in Prepare()
97 TF_LITE_ENSURE_EQ(context, NumDimensions(weights), 1); in Prepare()
98 TF_LITE_ENSURE_EQ(context, weights->type, kTfLiteFloat32); in Prepare()
[all …]
Dbatch_to_space_nd.cc62 TF_LITE_ENSURE_EQ(context, NumDimensions(op_context->block_shape), in ResizeOutputTensor()
64 TF_LITE_ENSURE_EQ(context, op_context->block_shape->dims->data[0], in ResizeOutputTensor()
66 TF_LITE_ENSURE_EQ(context, NumDimensions(op_context->crops), in ResizeOutputTensor()
71 TF_LITE_ENSURE_EQ(context, crops[0], 0); in ResizeOutputTensor()
72 TF_LITE_ENSURE_EQ(context, crops[1], 0); in ResizeOutputTensor()
73 TF_LITE_ENSURE_EQ(context, crops[2], 0); in ResizeOutputTensor()
74 TF_LITE_ENSURE_EQ(context, crops[3], 0); in ResizeOutputTensor()
77 TF_LITE_ENSURE_EQ(context, in ResizeOutputTensor()
96 TF_LITE_ENSURE_EQ(context, NumInputs(node), 3); in Prepare()
97 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
[all …]
Dhashtable_lookup.cc60 TF_LITE_ENSURE_EQ(context, NumInputs(node), 3); in Prepare()
61 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 2); in Prepare()
64 TF_LITE_ENSURE_EQ(context, NumDimensions(lookup), 1); in Prepare()
65 TF_LITE_ENSURE_EQ(context, lookup->type, kTfLiteInt32); in Prepare()
68 TF_LITE_ENSURE_EQ(context, NumDimensions(key), 1); in Prepare()
69 TF_LITE_ENSURE_EQ(context, key->type, kTfLiteInt32); in Prepare()
73 TF_LITE_ENSURE_EQ(context, SizeOfDimension(key, 0), in Prepare()
76 TF_LITE_ENSURE_EQ(context, NumDimensions(value), 1); in Prepare()
80 TF_LITE_ENSURE_EQ(context, hits->type, kTfLiteUInt8); in Prepare()
85 TF_LITE_ENSURE_EQ(context, value->type, output->type); in Prepare()
Dconcatenation.cc59 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()
71 TF_LITE_ENSURE_EQ(context, t->params.zero_point, t0->params.zero_point); in Prepare()
72 TF_LITE_ENSURE_EQ(context, t->params.scale, t0->params.scale); in Prepare()
78 TF_LITE_ENSURE_EQ(context, t->dims->data[d], t0->dims->data[d]); in Prepare()
89 TF_LITE_ENSURE_EQ(context, output->type, input_type); in Prepare()
91 TF_LITE_ENSURE_EQ(context, output->params.zero_point, in Prepare()
93 TF_LITE_ENSURE_EQ(context, output->params.scale, t0->params.scale); in Prepare()
Dspace_to_batch_nd.cc62 TF_LITE_ENSURE_EQ(context, NumDimensions(op_context->block_shape), in ResizeOutputTensor()
64 TF_LITE_ENSURE_EQ(context, op_context->block_shape->dims->data[0], in ResizeOutputTensor()
66 TF_LITE_ENSURE_EQ(context, NumDimensions(op_context->paddings), in ResizeOutputTensor()
76 TF_LITE_ENSURE_EQ(context, final_dim_size % block_shape[dim], 0); in ResizeOutputTensor()
91 TF_LITE_ENSURE_EQ(context, NumInputs(node), 3); in Prepare()
92 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
95 TF_LITE_ENSURE_EQ(context, NumDimensions(op_context.input), in Prepare()
97 TF_LITE_ENSURE_EQ(context, op_context.input->type, op_context.output->type); in Prepare()
Ddepthwise_conv.cc90 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
93 TF_LITE_ENSURE_EQ(context, NumDimensions(input), 4); in Prepare()
94 TF_LITE_ENSURE_EQ(context, NumDimensions(filter), 4); in Prepare()
98 TF_LITE_ENSURE_EQ(context, in Prepare()
105 TF_LITE_ENSURE_EQ(context, output->type, data_type); in Prepare()
106 TF_LITE_ENSURE_EQ(context, filter->type, data_type); in Prepare()
111 TF_LITE_ENSURE_EQ(context, bias->type, kTfLiteInt32); in Prepare()
112 TF_LITE_ENSURE_EQ(context, bias->params.zero_point, 0); in Prepare()
114 TF_LITE_ENSURE_EQ(context, bias->type, data_type); in Prepare()
116 TF_LITE_ENSURE_EQ(context, NumDimensions(bias), 1); in Prepare()
[all …]
Dstrided_slice.cc143 TF_LITE_ENSURE_EQ(context, NumInputs(node), 4); in Prepare()
144 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
149 TF_LITE_ENSURE_EQ(context, NumDimensions(op_context.begin), 1); in Prepare()
150 TF_LITE_ENSURE_EQ(context, NumDimensions(op_context.end), 1); in Prepare()
151 TF_LITE_ENSURE_EQ(context, NumDimensions(op_context.strides), 1); in Prepare()
152 TF_LITE_ENSURE_EQ(context, op_context.input->type, op_context.output->type); in Prepare()
155 TF_LITE_ENSURE_EQ(context, op_context.begin->type, kTfLiteInt32); in Prepare()
156 TF_LITE_ENSURE_EQ(context, op_context.end->type, kTfLiteInt32); in Prepare()
157 TF_LITE_ENSURE_EQ(context, op_context.strides->type, kTfLiteInt32); in Prepare()
Ddiv.cc41 TF_LITE_ENSURE_EQ(context, NumInputs(node), 2); in Prepare()
42 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
48 TF_LITE_ENSURE_EQ(context, NumDimensions(input1), NumDimensions(input2)); in Prepare()
50 TF_LITE_ENSURE_EQ(context, SizeOfDimension(input1, i), in Prepare()
54 TF_LITE_ENSURE_EQ(context, input1->type, output->type); in Prepare()
55 TF_LITE_ENSURE_EQ(context, input2->type, output->type); in Prepare()
Dspace_to_depth.cc42 TF_LITE_ENSURE_EQ(context, NumInputs(node), 1); in Prepare()
43 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
48 TF_LITE_ENSURE_EQ(context, NumDimensions(input), 4); in Prepare()
54 TF_LITE_ENSURE_EQ(context, input->type, output->type); in Prepare()
62 TF_LITE_ENSURE_EQ(context, input_height, output_height * block_size); in Prepare()
63 TF_LITE_ENSURE_EQ(context, input_width, output_width * block_size); in Prepare()
Dsub.cc41 TF_LITE_ENSURE_EQ(context, NumInputs(node), 2); in Prepare()
42 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
48 TF_LITE_ENSURE_EQ(context, NumDimensions(input1), NumDimensions(input2)); in Prepare()
50 TF_LITE_ENSURE_EQ(context, SizeOfDimension(input1, i), in Prepare()
54 TF_LITE_ENSURE_EQ(context, input1->type, output->type); in Prepare()
55 TF_LITE_ENSURE_EQ(context, input2->type, output->type); in Prepare()
Dresize_bilinear.cc51 TF_LITE_ENSURE_EQ(context, NumInputs(node), 2); in Prepare()
52 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
59 TF_LITE_ENSURE_EQ(context, NumDimensions(input), 4); in Prepare()
60 TF_LITE_ENSURE_EQ(context, NumDimensions(size), 1); in Prepare()
63 TF_LITE_ENSURE_EQ(context, input->type, kTfLiteFloat32); in Prepare()
64 TF_LITE_ENSURE_EQ(context, size->type, kTfLiteInt32); in Prepare()
Dactivations.cc56 TF_LITE_ENSURE_EQ(context, NumInputs(node), 1); in GenericPrepare()
57 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in GenericPrepare()
60 TF_LITE_ENSURE_EQ(context, input->type, output->type); in GenericPrepare()
69 TF_LITE_ENSURE_EQ(context, NumInputs(node), 1); in SigmoidPrepare()
70 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in SigmoidPrepare()
73 TF_LITE_ENSURE_EQ(context, input->type, output->type); in SigmoidPrepare()
76 TF_LITE_ENSURE_EQ(context, output->params.zero_point, 0); in SigmoidPrepare()
100 TF_LITE_ENSURE_EQ(context, NumInputs(node), 1); in SoftmaxPrepare()
101 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in SoftmaxPrepare()
104 TF_LITE_ENSURE_EQ(context, input->type, output->type); in SoftmaxPrepare()
[all …]
Dgather.cc33 TF_LITE_ENSURE_EQ(context, NumInputs(node), 2); in Prepare()
34 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
42 TF_LITE_ENSURE_EQ(context, positions->type, kTfLiteInt32); in Prepare()
44 TF_LITE_ENSURE_EQ(context, input->type, output->type); in Prepare()
48 TF_LITE_ENSURE_EQ(context, params->axis, 0); in Prepare()
59 TF_LITE_ENSURE_EQ(context, NumDimensions(input), 1); in Prepare()
Dl2norm.cc40 TF_LITE_ENSURE_EQ(context, NumInputs(node), 1); in Prepare()
41 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
49 TF_LITE_ENSURE_EQ(context, output->type, kTfLiteFloat32); in Prepare()
50 TF_LITE_ENSURE_EQ(context, input->type, output->type); in Prepare()
54 TF_LITE_ENSURE_EQ(context, params->activation, kTfLiteActNone); in Prepare()
Dlocal_response_norm.cc38 TF_LITE_ENSURE_EQ(context, NumInputs(node), 1); in Prepare()
39 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
44 TF_LITE_ENSURE_EQ(context, NumDimensions(input), 4); in Prepare()
46 TF_LITE_ENSURE_EQ(context, output->type, kTfLiteFloat32); in Prepare()
47 TF_LITE_ENSURE_EQ(context, input->type, output->type); in Prepare()
Dpad.cc55 TF_LITE_ENSURE_EQ(context, SizeOfDimension(op_context->paddings, 0), in ResizeOutputTensor()
57 TF_LITE_ENSURE_EQ(context, SizeOfDimension(op_context->paddings, 1), 2); in ResizeOutputTensor()
79 TF_LITE_ENSURE_EQ(context, NumInputs(node), 2); in Prepare()
80 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
83 TF_LITE_ENSURE_EQ(context, op_context.input->type, op_context.output->type); in Prepare()
86 TF_LITE_ENSURE_EQ(context, op_context.dims, 4); in Prepare()
Dtranspose.cc51 TF_LITE_ENSURE_EQ(context, NumDimensions(op_context->perm), 1); in ResizeOutputTensor()
52 TF_LITE_ENSURE_EQ(context, op_context->perm->dims->data[0], dims); in ResizeOutputTensor()
69 TF_LITE_ENSURE_EQ(context, NumInputs(node), 2); in Prepare()
70 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
77 TF_LITE_ENSURE_EQ(context, op_context.input->type, op_context.output->type); in Prepare()
Dconv.cc117 TF_LITE_ENSURE_EQ(context, node->outputs->size, 1); in Prepare()
122 TF_LITE_ENSURE_EQ(context, input->dims->size, 4); in Prepare()
123 TF_LITE_ENSURE_EQ(context, filter->dims->size, 4); in Prepare()
125 TF_LITE_ENSURE_EQ(context, input->dims->data[3], filter->dims->data[3]); in Prepare()
131 TF_LITE_ENSURE_EQ(context, output->type, data_type); in Prepare()
132 TF_LITE_ENSURE_EQ(context, filter->type, data_type); in Prepare()
143 TF_LITE_ENSURE_EQ(context, bias->type, kTfLiteInt32); in Prepare()
144 TF_LITE_ENSURE_EQ(context, bias->params.zero_point, 0); in Prepare()
146 TF_LITE_ENSURE_EQ(context, bias->type, data_type); in Prepare()
148 TF_LITE_ENSURE_EQ(context, bias->dims->size, 1); in Prepare()
[all …]
Dembedding_lookup.cc51 TF_LITE_ENSURE_EQ(context, NumInputs(node), 2); in Prepare()
52 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
55 TF_LITE_ENSURE_EQ(context, NumDimensions(lookup), 1); in Prepare()
56 TF_LITE_ENSURE_EQ(context, lookup->type, kTfLiteInt32); in Prepare()
Dtopk_v2.cc35 TF_LITE_ENSURE_EQ(context, top_k->type, kTfLiteInt32); in ResizeOutput()
37 TF_LITE_ENSURE_EQ(context, NumDimensions(top_k), 1); in ResizeOutput()
38 TF_LITE_ENSURE_EQ(context, NumElements(top_k), 1); in ResizeOutput()
162 TF_LITE_ENSURE_EQ(context, NumInputs(node), 2); in Prepare()
163 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 2); in Prepare()
167 TF_LITE_ENSURE_EQ(context, input->type, output_values->type); in Prepare()
170 TF_LITE_ENSURE_EQ(context, top_k->type, kTfLiteInt32); in Prepare()
Dskip_gram.cc49 TF_LITE_ENSURE_EQ(context, NumInputs(node), 1); in Prepare()
50 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in Prepare()
52 TF_LITE_ENSURE_EQ(context, GetInput(context, node, 0)->type, kTfLiteString); in Prepare()
53 TF_LITE_ENSURE_EQ(context, GetOutput(context, node, 0)->type, kTfLiteString); in Prepare()
Dpooling.cc69 TF_LITE_ENSURE_EQ(context, NumInputs(node), 1); in GenericPrepare()
70 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); in GenericPrepare()
73 TF_LITE_ENSURE_EQ(context, NumDimensions(input), 4); in GenericPrepare()
74 TF_LITE_ENSURE_EQ(context, input->type, output->type); in GenericPrepare()
104 TF_LITE_ENSURE_EQ(context, input->params.scale, output->params.scale); in GenericPrepare()
105 TF_LITE_ENSURE_EQ(context, input->params.zero_point, in GenericPrepare()
110 TF_LITE_ENSURE_EQ(context, input->type, kTfLiteFloat32); in GenericPrepare()
/external/tensorflow/tensorflow/contrib/lite/models/smartreply/ops/
Dpredict.cc76 TF_LITE_ENSURE_EQ(context, node->inputs->size, 4); in Prepare()
77 TF_LITE_ENSURE_EQ(context, node->outputs->size, 2); in Prepare()
83 TF_LITE_ENSURE_EQ(context, lookup->type, kTfLiteInt32); in Prepare()
84 TF_LITE_ENSURE_EQ(context, model_key->type, kTfLiteInt32); in Prepare()
85 TF_LITE_ENSURE_EQ(context, model_label->type, kTfLiteInt32); in Prepare()
86 TF_LITE_ENSURE_EQ(context, model_weight->type, kTfLiteFloat32); in Prepare()
87 TF_LITE_ENSURE_EQ(context, lookup->dims->size, 1); in Prepare()
88 TF_LITE_ENSURE_EQ(context, model_key->dims->size, 1); in Prepare()
89 TF_LITE_ENSURE_EQ(context, model_label->dims->size, 2); in Prepare()
90 TF_LITE_ENSURE_EQ(context, model_weight->dims->size, 2); in Prepare()
[all …]

12