/external/tensorflow/tensorflow/contrib/lite/kernels/ |
D | unidirectional_sequence_lstm.cc | 88 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 …]
|
D | lstm.cc | 88 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 …]
|
D | embedding_lookup_sparse.cc | 81 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 …]
|
D | batch_to_space_nd.cc | 62 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 …]
|
D | hashtable_lookup.cc | 60 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()
|
D | concatenation.cc | 59 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()
|
D | space_to_batch_nd.cc | 62 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()
|
D | depthwise_conv.cc | 90 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 …]
|
D | strided_slice.cc | 143 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()
|
D | div.cc | 41 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()
|
D | space_to_depth.cc | 42 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()
|
D | sub.cc | 41 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()
|
D | resize_bilinear.cc | 51 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()
|
D | activations.cc | 56 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 …]
|
D | gather.cc | 33 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()
|
D | l2norm.cc | 40 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()
|
D | local_response_norm.cc | 38 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()
|
D | pad.cc | 55 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()
|
D | transpose.cc | 51 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()
|
D | conv.cc | 117 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 …]
|
D | embedding_lookup.cc | 51 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()
|
D | topk_v2.cc | 35 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()
|
D | skip_gram.cc | 49 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()
|
D | pooling.cc | 69 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/ |
D | predict.cc | 76 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 …]
|