Home
last modified time | relevance | path

Searched refs:GetInput (Results 1 – 25 of 100) sorted by relevance

1234

/external/tensorflow/tensorflow/core/kernels/
Dparse_tensor_test.cc84 test::ExpectTensorEqual<Eigen::half>(parse_output, GetInput(0)); in TEST_F()
93 test::ExpectTensorEqual<float>(parse_output, GetInput(0)); in TEST_F()
102 test::ExpectTensorEqual<double>(parse_output, GetInput(0)); in TEST_F()
111 test::ExpectTensorEqual<int64>(parse_output, GetInput(0)); in TEST_F()
120 test::ExpectTensorEqual<int32>(parse_output, GetInput(0)); in TEST_F()
129 test::ExpectTensorEqual<int16>(parse_output, GetInput(0)); in TEST_F()
138 test::ExpectTensorEqual<int8>(parse_output, GetInput(0)); in TEST_F()
147 test::ExpectTensorEqual<uint16>(parse_output, GetInput(0)); in TEST_F()
156 test::ExpectTensorEqual<uint8>(parse_output, GetInput(0)); in TEST_F()
166 test::ExpectTensorEqual<complex64>(parse_output, GetInput(0)); in TEST_F()
[all …]
Dfake_quant_ops_test.cc379 auto input_flat = GetInput(0).flat<float>(); in TEST_F()
409 auto input_flat = GetInput(0).flat<float>(); in TEST_F()
440 auto input_flat = GetInput(0).flat<float>(); in TEST_F()
471 auto input_flat = GetInput(0).flat<float>(); in TEST_F()
669 auto in_flat = GetInput(0).flat<float>(); in TEST_F()
714 auto in_flat = GetInput(0).flat<float>(); in TEST_F()
758 auto in_flat = GetInput(0).flat<float>(); in TEST_F()
803 auto in_flat = GetInput(0).flat<float>(); in TEST_F()
848 auto in_flat = GetInput(0).flat<float>(); in TEST_F()
1245 auto grad_flat = GetInput(0).flat<float>(); in TEST_F()
[all …]
/external/tensorflow/tensorflow/lite/kernels/
Dlstm.cc138 GetInput(context, node, kInputToForgetWeightsTensor); in CheckInputTensorDimensions()
144 GetInput(context, node, kInputToCellWeightsTensor); in CheckInputTensorDimensions()
160 GetInput(context, node, kRecurrentToForgetWeightsTensor); in CheckInputTensorDimensions()
168 GetInput(context, node, kRecurrentToCellWeightsTensor); in CheckInputTensorDimensions()
225 GetInput(context, node, kForgetGateBiasTensor); in CheckInputTensorDimensions()
229 const TfLiteTensor* cell_bias = GetInput(context, node, kCellGateBiasTensor); in CheckInputTensorDimensions()
234 GetInput(context, node, kOutputGateBiasTensor); in CheckInputTensorDimensions()
275 GetInput(context, node, kForgetLayerNormCoefficientsTensor); in CheckInputTensorDimensions()
282 GetInput(context, node, kCellLayerNormCoefficientsTensor); in CheckInputTensorDimensions()
289 GetInput(context, node, kOutputLayerNormCoefficientsTensor); in CheckInputTensorDimensions()
[all …]
Dembedding_lookup_sparse.cc84 const TfLiteTensor* ids = GetInput(context, node, 0); in Prepare()
88 const TfLiteTensor* indices = GetInput(context, node, 1); in Prepare()
92 const TfLiteTensor* shape = GetInput(context, node, 2); in Prepare()
96 const TfLiteTensor* weights = GetInput(context, node, 3); in Prepare()
105 const TfLiteTensor* value = GetInput(context, node, 4); in Prepare()
142 const TfLiteTensor* ids = GetInput(context, node, 0); in Eval()
143 const TfLiteTensor* indices = GetInput(context, node, 1); in Eval()
144 const TfLiteTensor* dense_shape = GetInput(context, node, 2); in Eval()
145 const TfLiteTensor* weights = GetInput(context, node, 3); in Eval()
146 const TfLiteTensor* value = GetInput(context, node, 4); in Eval()
Dcomparisons.cc36 const TfLiteTensor* input1 = GetInput(context, node, kInputTensor1); in ComparisonPrepare()
37 const TfLiteTensor* input2 = GetInput(context, node, kInputTensor2); in ComparisonPrepare()
127 const TfLiteTensor* input1 = GetInput(context, node, kInputTensor1); in EqualEval()
128 const TfLiteTensor* input2 = GetInput(context, node, kInputTensor2); in EqualEval()
163 const TfLiteTensor* input1 = GetInput(context, node, kInputTensor1); in NotEqualEval()
164 const TfLiteTensor* input2 = GetInput(context, node, kInputTensor2); in NotEqualEval()
198 const TfLiteTensor* input1 = GetInput(context, node, kInputTensor1); in GreaterEval()
199 const TfLiteTensor* input2 = GetInput(context, node, kInputTensor2); in GreaterEval()
230 const TfLiteTensor* input1 = GetInput(context, node, kInputTensor1); in GreaterEqualEval()
231 const TfLiteTensor* input2 = GetInput(context, node, kInputTensor2); in GreaterEqualEval()
[all …]
Dunidirectional_sequence_lstm.cc120 GetInput(context, node, kInputToForgetWeightsTensor); in CheckInputTensorDimensions()
126 GetInput(context, node, kInputToCellWeightsTensor); in CheckInputTensorDimensions()
142 GetInput(context, node, kRecurrentToForgetWeightsTensor); in CheckInputTensorDimensions()
150 GetInput(context, node, kRecurrentToCellWeightsTensor); in CheckInputTensorDimensions()
208 GetInput(context, node, kForgetGateBiasTensor); in CheckInputTensorDimensions()
212 const TfLiteTensor* cell_bias = GetInput(context, node, kCellGateBiasTensor); in CheckInputTensorDimensions()
217 GetInput(context, node, kOutputGateBiasTensor); in CheckInputTensorDimensions()
260 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare()
271 GetInput(context, node, kInputToOutputWeightsTensor); in Prepare()
277 GetInput(context, node, kRecurrentToOutputWeightsTensor); in Prepare()
[all …]
Dbidirectional_sequence_lstm.cc179 GetInput(context, node, input_to_forget_weights_tensor); in CheckLstmTensorDimensionsAndTypes()
197 GetInput(context, node, input_to_cell_weights_tensor); in CheckLstmTensorDimensionsAndTypes()
205 GetInput(context, node, input_to_output_weights_tensor); in CheckLstmTensorDimensionsAndTypes()
225 GetInput(context, node, recurrent_to_forget_weights_tensor); in CheckLstmTensorDimensionsAndTypes()
235 GetInput(context, node, recurrent_to_cell_weights_tensor); in CheckLstmTensorDimensionsAndTypes()
302 GetInput(context, node, forget_gate_bias_tensor); in CheckLstmTensorDimensionsAndTypes()
308 GetInput(context, node, cell_gate_bias_tensor); in CheckLstmTensorDimensionsAndTypes()
314 GetInput(context, node, output_gate_bias_tensor); in CheckLstmTensorDimensionsAndTypes()
398 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare()
407 GetInput(context, node, kFwInputToOutputWeightsTensor); in Prepare()
[all …]
Dsparse_to_dense.cc150 const TfLiteTensor* indices = GetInput(context, node, kIndicesTensor); in Prepare()
152 GetInput(context, node, kOutputShapeTensor); in Prepare()
153 const TfLiteTensor* values = GetInput(context, node, kValueInputTensor); in Prepare()
155 GetInput(context, node, kDefaultValueTensor); in Prepare()
192 const TfLiteTensor* indices = GetInput(context, node, kIndicesTensor); in SparseToDenseImpl()
194 GetInput(context, node, kOutputShapeTensor); in SparseToDenseImpl()
195 const TfLiteTensor* values = GetInput(context, node, kValueInputTensor); in SparseToDenseImpl()
197 GetInput(context, node, kDefaultValueTensor); in SparseToDenseImpl()
220 const TfLiteTensor* indices = GetInput(context, node, kIndicesTensor); in Eval()
221 const TfLiteTensor* values = GetInput(context, node, kValueInputTensor); in Eval()
Dselect.cc37 GetInput(context, node, kInputTensorCondition); in SelectPrepare()
38 const TfLiteTensor* input_x = GetInput(context, node, kInputTensorX); in SelectPrepare()
39 const TfLiteTensor* input_y = GetInput(context, node, kInputTensorY); in SelectPrepare()
66 GetInput(context, node, kInputTensorCondition); in SelectEval()
67 const TfLiteTensor* input_x = GetInput(context, node, kInputTensorX); in SelectEval()
68 const TfLiteTensor* input_y = GetInput(context, node, kInputTensorY); in SelectEval()
Dbasic_rnn.cc64 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare()
65 const TfLiteTensor* input_weights = GetInput(context, node, kWeightsTensor); in Prepare()
67 GetInput(context, node, kRecurrentWeightsTensor); in Prepare()
68 const TfLiteTensor* bias = GetInput(context, node, kBiasTensor); in Prepare()
70 GetInput(context, node, kHiddenStateTensor); in Prepare()
221 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval()
222 const TfLiteTensor* input_weights = GetInput(context, node, kWeightsTensor); in Eval()
224 GetInput(context, node, kRecurrentWeightsTensor); in Eval()
225 const TfLiteTensor* bias = GetInput(context, node, kBiasTensor); in Eval()
Dhashtable_lookup.cc62 const TfLiteTensor* lookup = GetInput(context, node, 0); in Prepare()
66 const TfLiteTensor* key = GetInput(context, node, 1); in Prepare()
70 const TfLiteTensor* value = GetInput(context, node, 2); in Prepare()
104 const TfLiteTensor* lookup = GetInput(context, node, 0); in Eval()
105 const TfLiteTensor* key = GetInput(context, node, 1); in Eval()
106 const TfLiteTensor* value = GetInput(context, node, 2); in Eval()
Dlsh_projection.cc79 const TfLiteTensor* hash = GetInput(context, node, 0); in Resize()
84 const TfLiteTensor* input = GetInput(context, node, 1); in Resize()
88 const TfLiteTensor* weight = GetInput(context, node, 2); in Resize()
175 const TfLiteTensor* hash = GetInput(context, node, 0); in Eval()
176 const TfLiteTensor* input = GetInput(context, node, 1); in Eval()
178 NumInputs(node) == 2 ? nullptr : GetInput(context, node, 2); in Eval()
Drange.cc80 const TfLiteTensor* start = GetInput(context, node, kStartTensor); in Prepare()
81 const TfLiteTensor* limit = GetInput(context, node, kLimitTensor); in Prepare()
82 const TfLiteTensor* delta = GetInput(context, node, kDeltaTensor); in Prepare()
133 const TfLiteTensor* start = GetInput(context, node, kStartTensor); in Eval()
134 const TfLiteTensor* limit = GetInput(context, node, kLimitTensor); in Eval()
135 const TfLiteTensor* delta = GetInput(context, node, kDeltaTensor); in Eval()
Dunidirectional_sequence_rnn.cc59 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare()
60 const TfLiteTensor* input_weights = GetInput(context, node, kWeightsTensor); in Prepare()
62 GetInput(context, node, kRecurrentWeightsTensor); in Prepare()
63 const TfLiteTensor* bias = GetInput(context, node, kBiasTensor); in Prepare()
65 GetInput(context, node, kHiddenStateTensor); in Prepare()
291 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval()
292 const TfLiteTensor* input_weights = GetInput(context, node, kWeightsTensor); in Eval()
294 GetInput(context, node, kRecurrentWeightsTensor); in Eval()
295 const TfLiteTensor* bias = GetInput(context, node, kBiasTensor); in Eval()
298 const_cast<TfLiteTensor*>(GetInput(context, node, kHiddenStateTensor)); in Eval()
Dreshape.cc33 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in ResizeOutput()
64 const TfLiteTensor* shape = GetInput(context, node, kShapeTensor); in GetOutputShapeFromTensor()
97 const TfLiteTensor* shape = GetInput(context, node, kShapeTensor); in ShapeIsVector()
120 IsConstantTensor(GetInput(context, node, kShapeTensor))) { in Prepare()
131 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval()
Dtile.cc47 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in ResizeOutput()
49 const TfLiteTensor* multipliers = GetInput(context, node, kInputMultipliers); in ResizeOutput()
141 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare()
146 const TfLiteTensor* multipliers = GetInput(context, node, kInputMultipliers); in Prepare()
164 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval()
166 const TfLiteTensor* multipliers = GetInput(context, node, kInputMultipliers); in Eval()
Dreverse_sequence.cc36 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare()
37 const TfLiteTensor* seq_lengths = GetInput(context, node, kSeqLengthsTensor); in Prepare()
65 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in ReverseSequenceImpl()
67 GetInput(context, node, kSeqLengthsTensor); in ReverseSequenceImpl()
99 GetInput(context, node, kSeqLengthsTensor); in ReverseSequenceHelper()
Dbidirectional_sequence_rnn.cc96 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare()
98 GetInput(context, node, kFwWeightsTensor); in Prepare()
100 GetInput(context, node, kFwRecurrentWeightsTensor); in Prepare()
101 const TfLiteTensor* fw_bias = GetInput(context, node, kFwBiasTensor); in Prepare()
103 GetInput(context, node, kFwHiddenStateTensor); in Prepare()
105 GetInput(context, node, kBwWeightsTensor); in Prepare()
107 GetInput(context, node, kBwRecurrentWeightsTensor); in Prepare()
108 const TfLiteTensor* bw_bias = GetInput(context, node, kBwBiasTensor); in Prepare()
110 GetInput(context, node, kBwHiddenStateTensor); in Prepare()
595 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval()
[all …]
Dslice.cc109 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare()
110 const TfLiteTensor* begin = GetInput(context, node, kBeginTensor); in Prepare()
111 const TfLiteTensor* size = GetInput(context, node, kSizeTensor); in Prepare()
137 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval()
138 const TfLiteTensor* begin = GetInput(context, node, kBeginTensor); in Eval()
139 const TfLiteTensor* size = GetInput(context, node, kSizeTensor); in Eval()
Dexpand_dims.cc75 const TfLiteTensor* input = GetInput(context, node, kInput); in Prepare()
76 const TfLiteTensor* axis = GetInput(context, node, kAxis); in Prepare()
91 const TfLiteTensor* input = GetInput(context, node, kInput); in Eval()
93 const TfLiteTensor* axis = GetInput(context, node, kAxis); in Eval()
Dfill.cc72 const TfLiteTensor* dims = GetInput(context, node, kDimsTensor); in Prepare()
73 const TfLiteTensor* value = GetInput(context, node, kValueTensor); in Prepare()
97 const TfLiteTensor* value = GetInput(context, node, kValueTensor); in Eval()
102 const TfLiteTensor* dims = GetInput(context, node, kDimsTensor); in Eval()
Dlogical.cc54 const TfLiteTensor* input1 = GetInput(context, node, kInputTensor1); in Prepare()
55 const TfLiteTensor* input2 = GetInput(context, node, kInputTensor2); in Prepare()
84 const TfLiteTensor* input1 = GetInput(context, node, kInputTensor1); in LogicalImpl()
85 const TfLiteTensor* input2 = GetInput(context, node, kInputTensor2); in LogicalImpl()
Dtranspose_conv.cc141 GetInput(context, node, kOutputShapeTensor); in Prepare()
142 const TfLiteTensor* weights = GetInput(context, node, kWeightsTensor); in Prepare()
143 const TfLiteTensor* input = GetInput(context, node, kDataInputTensor); in Prepare()
177 GetInput(context, node, kOutputShapeTensor); in Eval()
178 const TfLiteTensor* weights = GetInput(context, node, kWeightsTensor); in Eval()
179 const TfLiteTensor* input = GetInput(context, node, kDataInputTensor); in Eval()
Dreverse.cc36 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare()
37 const TfLiteTensor* axis = GetInput(context, node, kAxisTensor); in Prepare()
68 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval()
69 const TfLiteTensor* axis_tensor = GetInput(context, node, kAxisTensor); in Eval()
/external/tensorflow/tensorflow/contrib/lite/kernels/
Dlsh_projection.cc79 const TfLiteTensor* hash = GetInput(context, node, 0); in Resize()
84 const TfLiteTensor* input = GetInput(context, node, 1); in Resize()
88 const TfLiteTensor* weight = GetInput(context, node, 2); in Resize()
175 const TfLiteTensor* hash = GetInput(context, node, 0); in Eval()
176 const TfLiteTensor* input = GetInput(context, node, 1); in Eval()
178 NumInputs(node) == 2 ? nullptr : GetInput(context, node, 2); in Eval()

1234