| /external/tensorflow/tensorflow/core/kernels/ |
| H A D | parse_tensor_test.cc | 84 test::ExpectTensorEqual<Eigen::half>(parse_output, GetInput(0));
93 test::ExpectTensorEqual<float>(parse_output, GetInput(0));
102 test::ExpectTensorEqual<double>(parse_output, GetInput(0));
111 test::ExpectTensorEqual<int64>(parse_output, GetInput(0));
120 test::ExpectTensorEqual<int32>(parse_output, GetInput(0));
129 test::ExpectTensorEqual<int16>(parse_output, GetInput(0));
138 test::ExpectTensorEqual<int8>(parse_output, GetInput(0));
147 test::ExpectTensorEqual<uint16>(parse_output, GetInput(0));
156 test::ExpectTensorEqual<uint8>(parse_output, GetInput(0));
166 test::ExpectTensorEqual<complex64>(parse_output, GetInput(
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| H A D | embedding_lookup.cc | 54 TfLiteTensor* lookup = GetInput(context, node, 0);
58 TfLiteTensor* value = GetInput(context, node, 1);
74 TfLiteTensor* lookup = GetInput(context, node, 0);
75 TfLiteTensor* value = GetInput(context, node, 1);
|
| H A D | embedding_lookup_sparse.cc | 84 TfLiteTensor* ids = GetInput(context, node, 0);
88 TfLiteTensor* indices = GetInput(context, node, 1);
92 TfLiteTensor* shape = GetInput(context, node, 2);
96 TfLiteTensor* weights = GetInput(context, node, 3);
105 TfLiteTensor* value = GetInput(context, node, 4);
142 TfLiteTensor* ids = GetInput(context, node, 0);
143 TfLiteTensor* indices = GetInput(context, node, 1);
144 TfLiteTensor* dense_shape = GetInput(context, node, 2);
145 TfLiteTensor* weights = GetInput(context, node, 3);
146 TfLiteTensor* value = GetInput(contex
[all...] |
| H A D | hashtable_lookup.cc | 63 TfLiteTensor* lookup = GetInput(context, node, 0);
67 TfLiteTensor* key = GetInput(context, node, 1);
71 TfLiteTensor* value = GetInput(context, node, 2);
105 TfLiteTensor* lookup = GetInput(context, node, 0);
106 TfLiteTensor* key = GetInput(context, node, 1);
107 TfLiteTensor* value = GetInput(context, node, 2);
|
| H A D | lsh_projection.cc | 80 TfLiteTensor* hash = GetInput(context, node, 0);
85 TfLiteTensor* input = GetInput(context, node, 1);
89 TfLiteTensor* weight = GetInput(context, node, 2);
176 TfLiteTensor* hash = GetInput(context, node, 0);
177 TfLiteTensor* input = GetInput(context, node, 1);
179 NumInputs(node) == 2 ? nullptr : GetInput(context, node, 2);
|
| H A D | div.cc | 44 TfLiteTensor* input1 = GetInput(context, node, kInputTensor1);
45 TfLiteTensor* input2 = GetInput(context, node, kInputTensor2);
85 TfLiteTensor* input1 = GetInput(context, node, kInputTensor1);
86 TfLiteTensor* input2 = GetInput(context, node, kInputTensor2);
|
| H A D | sub.cc | 44 TfLiteTensor* input1 = GetInput(context, node, kInputTensor1);
45 TfLiteTensor* input2 = GetInput(context, node, kInputTensor2);
85 TfLiteTensor* input1 = GetInput(context, node, kInputTensor1);
86 TfLiteTensor* input2 = GetInput(context, node, kInputTensor2);
|
| H A D | lstm.cc | 94 GetInput(context, node, kInputToForgetWeightsTensor);
100 GetInput(context, node, kInputToCellWeightsTensor);
116 GetInput(context, node, kRecurrentToForgetWeightsTensor);
124 GetInput(context, node, kRecurrentToCellWeightsTensor);
182 GetInput(context, node, kForgetGateBiasTensor);
186 TfLiteTensor* cell_bias = GetInput(context, node, kCellGateBiasTensor);
191 GetInput(context, node, kOutputGateBiasTensor);
231 TfLiteTensor* input = GetInput(context, node, kInputTensor);
237 GetInput(context, node, kInputToOutputWeightsTensor);
243 GetInput(contex
[all...] |
| H A D | unidirectional_sequence_lstm.cc | 94 GetInput(context, node, kInputToForgetWeightsTensor);
100 GetInput(context, node, kInputToCellWeightsTensor);
116 GetInput(context, node, kRecurrentToForgetWeightsTensor);
124 GetInput(context, node, kRecurrentToCellWeightsTensor);
182 GetInput(context, node, kForgetGateBiasTensor);
186 TfLiteTensor* cell_bias = GetInput(context, node, kCellGateBiasTensor);
191 GetInput(context, node, kOutputGateBiasTensor);
231 TfLiteTensor* input = GetInput(context, node, kInputTensor);
238 GetInput(context, node, kInputToOutputWeightsTensor);
244 GetInput(contex
[all...] |
| H A D | gather.cc | 38 TfLiteTensor* input = GetInput(context, node, kInputTensor);
39 TfLiteTensor* positions = GetInput(context, node, kInputPositions);
84 TfLiteTensor* input = GetInput(context, node, kInputTensor);
85 TfLiteTensor* positions = GetInput(context, node, kInputPositions);
|
| H A D | resize_bilinear.cc | 54 TfLiteTensor* input = GetInput(context, node, kInputTensor);
55 TfLiteTensor* size = GetInput(context, node, kSizeTensor);
81 TfLiteTensor* input = GetInput(context, node, kInputTensor);
83 TfLiteTensor* size = GetInput(context, node, kSizeTensor);
|
| H A D | l2norm.cc | 43 TfLiteTensor* input = GetInput(context, node, kInputTensor);
62 TfLiteTensor* input = GetInput(context, node, kInputTensor);
|
| H A D | reshape.cc | 38 TfLiteTensor* input = GetInput(context, node, kInputTensor);
73 TfLiteTensor* input = GetInput(context, node, kInputTensor);
|
| H A D | activations.cc | 58 TfLiteTensor* input = GetInput(context, node, 0);
71 TfLiteTensor* input = GetInput(context, node, 0);
102 TfLiteTensor* input = GetInput(context, node, 0);
127 TfLiteTensor* input = GetInput(context, node, 0);
145 TfLiteTensor* input = GetInput(context, node, 0);
165 TfLiteTensor* input = GetInput(context, node, 0);
183 TfLiteTensor* input = GetInput(context, node, 0);
204 TfLiteTensor* input = GetInput(context, node, 0);
301 TfLiteTensor* input = GetInput(context, node, 0);
|
| H A D | batch_to_space_nd.cc | 38 input = GetInput(context, node, 0);
39 block_shape = GetInput(context, node, 1);
40 crops = GetInput(context, node, 2);
|
| H A D | space_to_batch_nd.cc | 38 input = GetInput(context, node, 0);
39 block_shape = GetInput(context, node, 1);
40 paddings = GetInput(context, node, 2);
|
| H A D | depthwise_conv.cc | 86 TfLiteTensor* input = GetInput(context, node, kInputTensor);
87 TfLiteTensor* filter = GetInput(context, node, kFilterTensor);
109 bias = GetInput(context, node, kBiasTensor);
233 TfLiteTensor* input = GetInput(context, node, kInputTensor);
234 TfLiteTensor* filter = GetInput(context, node, kFilterTensor);
236 (NumInputs(node) == 3) ? GetInput(context, node, kBiasTensor) : nullptr;
|
| H A D | topk_v2.cc | 33 TfLiteTensor* top_k = GetInput(context, node, kInputTopK);
41 TfLiteTensor* input = GetInput(context, node, kInputTensor);
165 TfLiteTensor* input = GetInput(context, node, kInputTensor);
169 TfLiteTensor* top_k = GetInput(context, node, kInputTopK);
190 TfLiteTensor* top_k = GetInput(context, node, kInputTopK);
194 TfLiteTensor* input = GetInput(context, node, kInputTensor);
|
| H A D | local_response_norm.cc | 41 TfLiteTensor* input = GetInput(context, node, kInputTensor);
63 TfLiteTensor* input = GetInput(context, node, kInputTensor);
|
| H A D | skip_gram.cc | 52 TF_LITE_ENSURE_EQ(context, GetInput(context, node, 0)->type, kTfLiteString);
93 tflite::StringRef strref = tflite::GetString(GetInput(context, node, 0), 0);
|
| H A D | add.cc | 60 TfLiteTensor* input1 = GetInput(context, node, kInputTensor1);
61 TfLiteTensor* input2 = GetInput(context, node, kInputTensor2);
167 TfLiteTensor* input1 = GetInput(context, node, kInputTensor1);
168 TfLiteTensor* input2 = GetInput(context, node, kInputTensor2);
|
| H A D | exp.cc | 36 input = GetInput(context, node, 0);
|
| H A D | mul.cc | 60 TfLiteTensor* input1 = GetInput(context, node, kInputTensor1);
61 TfLiteTensor* input2 = GetInput(context, node, kInputTensor2);
152 TfLiteTensor* input1 = GetInput(context, node, kInputTensor1);
153 TfLiteTensor* input2 = GetInput(context, node, kInputTensor2);
|
| H A D | pad.cc | 38 input = GetInput(context, node, 0);
39 paddings = GetInput(context, node, 1);
|
| /external/tensorflow/tensorflow/contrib/lite/models/smartreply/ops/ |
| H A D | extract_feature.cc | 64 TfLiteTensor* input = GetInput(context, node, 0);
79 TfLiteTensor* input = GetInput(context, node, 0);
|