Home
last modified time | relevance | path

Searched refs:GetInputSafe (Results 1 – 25 of 111) sorted by relevance

12345

/external/tensorflow/tensorflow/lite/kernels/
Dnon_max_suppression.cc84 context, GetInputSafe(context, node, kInputTensorBoxes, &input_boxes)); in Prepare()
91 context, GetInputSafe(context, node, kInputTensorScores, &input_scores)); in Prepare()
99 GetInputSafe(context, node, kInputTensorMaxOutputSize, in Prepare()
113 GetInputSafe(context, node, kInputTensorIouThreshold, in Prepare()
119 GetInputSafe(context, node, kInputTensorScoreThreshold, in Prepare()
127 context, GetInputSafe(context, node, kInputTensorSigma, &input_sigma)); in Prepare()
206 context, GetInputSafe(context, node, kInputTensorBoxes, &input_boxes)); in Eval()
210 context, GetInputSafe(context, node, kInputTensorScores, &input_scores)); in Eval()
213 GetInputSafe(context, node, kInputTensorMaxOutputSize, in Eval()
220 GetInputSafe(context, node, kInputTensorIouThreshold, in Eval()
[all …]
Dembedding_lookup_sparse.cc88 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 0, &ids)); in Prepare()
93 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 1, &indices)); in Prepare()
98 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 2, &shape)); in Prepare()
103 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 3, &weights)); in Prepare()
113 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 4, &value)); in Prepare()
153 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 0, &ids)); in Eval()
155 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 1, &indices)); in Eval()
157 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 2, &dense_shape)); in Eval()
159 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 3, &weights)); in Eval()
161 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 4, &value)); in Eval()
Dhashtable_import.cc36 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputResourceIdTensor, in PrepareHashtableImport()
44 GetInputSafe(context, node, kKeyTensor, &key_tensor)); in PrepareHashtableImport()
47 GetInputSafe(context, node, kValueTensor, &value_tensor)); in PrepareHashtableImport()
60 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputResourceIdTensor, in EvalHashtableImport()
66 GetInputSafe(context, node, kKeyTensor, &key_tensor)); in EvalHashtableImport()
69 GetInputSafe(context, node, kValueTensor, &value_tensor)); in EvalHashtableImport()
Dunidirectional_sequence_gru.cc139 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInput, &input)); in Prepare()
148 GetInputSafe(context, node, kInputState, &input_state)); in Prepare()
156 GetInputSafe(context, node, kGateWeight, &gate_weight)); in Prepare()
164 GetInputSafe(context, node, kGateBias, &gate_bias)); in Prepare()
170 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kCandidateWeight, in Prepare()
180 context, GetInputSafe(context, node, kCandidateBias, &candidate_bias)); in Prepare()
235 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInput, &input)); in Eval()
238 GetInputSafe(context, node, kInputState, &input_state)); in Eval()
241 GetInputSafe(context, node, kGateWeight, &gate_weight)); in Eval()
244 GetInputSafe(context, node, kGateBias, &gate_bias)); in Eval()
[all …]
Dhashtable_find.cc37 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputResourceIdTensor, in PrepareHashtableFind()
44 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kDefaultValueTensor, in PrepareHashtableFind()
49 GetInputSafe(context, node, kKeyTensor, &key_tensor)); in PrepareHashtableFind()
64 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputResourceIdTensor, in EvalHashtableFind()
70 GetInputSafe(context, node, kKeyTensor, &key_tensor)); in EvalHashtableFind()
72 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kDefaultValueTensor, in EvalHashtableFind()
Dsparse_to_dense.cc148 GetInputSafe(context, node, kIndicesTensor, &indices)); in Prepare()
151 context, GetInputSafe(context, node, kOutputShapeTensor, &output_shape)); in Prepare()
154 GetInputSafe(context, node, kValueInputTensor, &values)); in Prepare()
156 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kDefaultValueTensor, in Prepare()
204 GetInputSafe(context, node, kIndicesTensor, &indices)); in SparseToDenseImpl()
207 context, GetInputSafe(context, node, kOutputShapeTensor, &output_shape)); in SparseToDenseImpl()
210 GetInputSafe(context, node, kValueInputTensor, &values)); in SparseToDenseImpl()
212 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kDefaultValueTensor, in SparseToDenseImpl()
259 GetInputSafe(context, node, kIndicesTensor, &indices)); in Eval()
262 GetInputSafe(context, node, kValueInputTensor, &values)); in Eval()
Dcomparisons.cc46 GetInputSafe(context, node, kInputTensor1, &input1)); in ComparisonPrepareCommon()
49 GetInputSafe(context, node, kInputTensor2, &input2)); in ComparisonPrepareCommon()
165 GetInputSafe(context, node, kInputTensor1, &input1)); in EqualEval()
168 GetInputSafe(context, node, kInputTensor2, &input2)); in EqualEval()
215 GetInputSafe(context, node, kInputTensor1, &input1)); in NotEqualEval()
218 GetInputSafe(context, node, kInputTensor2, &input2)); in NotEqualEval()
265 GetInputSafe(context, node, kInputTensor1, &input1)); in GreaterEval()
268 GetInputSafe(context, node, kInputTensor2, &input2)); in GreaterEval()
306 GetInputSafe(context, node, kInputTensor1, &input1)); in GreaterEqualEval()
309 GetInputSafe(context, node, kInputTensor2, &input2)); in GreaterEqualEval()
[all …]
Dhashtable_lookup.cc59 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 0, &lookup)); in Prepare()
64 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 1, &key)); in Prepare()
69 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 2, &value)); in Prepare()
108 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 0, &lookup)); in Eval()
110 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 1, &key)); in Eval()
112 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 2, &value)); in Eval()
Dunidirectional_sequence_lstm.cc94 context, GetInputSafe(context, node, lstm::full::kInputTensor, &input)); in PopulateQuantizedLstmParams8x8_16()
101 GetInputSafe(context, node, lstm::full::kInputToForgetWeightsTensor, in PopulateQuantizedLstmParams8x8_16()
104 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, in PopulateQuantizedLstmParams8x8_16()
110 GetInputSafe(context, node, lstm::full::kInputToOutputWeightsTensor, in PopulateQuantizedLstmParams8x8_16()
118 GetInputSafe(context, node, lstm::full::kRecurrentToForgetWeightsTensor, in PopulateQuantizedLstmParams8x8_16()
123 GetInputSafe(context, node, lstm::full::kRecurrentToCellWeightsTensor, in PopulateQuantizedLstmParams8x8_16()
128 GetInputSafe(context, node, lstm::full::kRecurrentToOutputWeightsTensor, in PopulateQuantizedLstmParams8x8_16()
440 GetInputSafe(context, node, lstm::full::kInputToForgetWeightsTensor, in CheckInputTensorDimensions()
447 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, in CheckInputTensorDimensions()
467 GetInputSafe(context, node, lstm::full::kRecurrentToForgetWeightsTensor, in CheckInputTensorDimensions()
[all …]
Dselect.cc65 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensorCondition, in SelectPrepare()
69 GetInputSafe(context, node, kInputTensorX, &input_x)); in SelectPrepare()
72 GetInputSafe(context, node, kInputTensorY, &input_y)); in SelectPrepare()
131 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensorCondition, in SelectEval()
135 GetInputSafe(context, node, kInputTensorX, &input_x)); in SelectEval()
138 GetInputSafe(context, node, kInputTensorY, &input_y)); in SelectEval()
Drange.cc87 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kStartTensor, &start)); in Prepare()
89 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kLimitTensor, &limit)); in Prepare()
91 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kDeltaTensor, &delta)); in Prepare()
139 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kStartTensor, &start)); in Eval()
141 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kLimitTensor, &limit)); in Eval()
143 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kDeltaTensor, &delta)); in Eval()
Dbidirectional_sequence_lstm.cc198 GetInputSafe(context, node, input_to_forget_weights_tensor, in CheckLstmTensorDimensionsAndTypes()
219 GetInputSafe(context, node, input_to_cell_weights_tensor, in CheckLstmTensorDimensionsAndTypes()
229 GetInputSafe(context, node, input_to_output_weights_tensor, in CheckLstmTensorDimensionsAndTypes()
251 context, GetInputSafe(context, node, recurrent_to_forget_weights_tensor, in CheckLstmTensorDimensionsAndTypes()
263 context, GetInputSafe(context, node, recurrent_to_cell_weights_tensor, in CheckLstmTensorDimensionsAndTypes()
333 GetInputSafe(context, node, forget_gate_bias_tensor, &forget_gate_bias)); in CheckLstmTensorDimensionsAndTypes()
339 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, cell_gate_bias_tensor, in CheckLstmTensorDimensionsAndTypes()
348 GetInputSafe(context, node, output_gate_bias_tensor, &output_gate_bias)); in CheckLstmTensorDimensionsAndTypes()
433 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Prepare()
443 GetInputSafe(context, node, kFwInputToOutputWeightsTensor, in Prepare()
[all …]
Dlstm.cc180 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in PopulateQuantizedLstmParams8x8_16()
186 GetInputSafe(context, node, kInputToForgetWeightsTensor, in PopulateQuantizedLstmParams8x8_16()
190 GetInputSafe(context, node, kInputToCellWeightsTensor, in PopulateQuantizedLstmParams8x8_16()
194 GetInputSafe(context, node, kInputToOutputWeightsTensor, in PopulateQuantizedLstmParams8x8_16()
201 GetInputSafe(context, node, kRecurrentToForgetWeightsTensor, in PopulateQuantizedLstmParams8x8_16()
205 GetInputSafe(context, node, kRecurrentToCellWeightsTensor, in PopulateQuantizedLstmParams8x8_16()
209 GetInputSafe(context, node, kRecurrentToOutputWeightsTensor, in PopulateQuantizedLstmParams8x8_16()
469 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in PopulateQuantizedLstmParams8x8_8()
474 GetInputSafe(context, node, kInputToForgetWeightsTensor, in PopulateQuantizedLstmParams8x8_8()
478 GetInputSafe(context, node, kInputToCellWeightsTensor, in PopulateQuantizedLstmParams8x8_8()
[all …]
Ddynamic_update_slice.cc45 GetInputSafe(context, node, kOperandTensor, &operand)); in Prepare()
48 GetInputSafe(context, node, kUpdateTensor, &update)); in Prepare()
50 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kStartIndicesTensor, in Prepare()
143 GetInputSafe(context, node, kOperandTensor, &operand)); in Eval()
146 GetInputSafe(context, node, kUpdateTensor, &update)); in Eval()
149 GetInputSafe(context, node, kStartIndicesTensor, &indice)); in Eval()
Dreverse_sequence.cc40 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Prepare()
43 context, GetInputSafe(context, node, kSeqLengthsTensor, &seq_lengths)); in Prepare()
74 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in ReverseSequenceImpl()
76 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kSeqLengthsTensor, in ReverseSequenceImpl()
111 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kSeqLengthsTensor, in ReverseSequenceHelper()
Dscatter_nd.cc78 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kIndices, &indices)); in Prepare()
80 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kUpdates, &updates)); in Prepare()
82 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kShape, &shape)); in Prepare()
185 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kIndices, &indices)); in Eval()
187 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kUpdates, &updates)); in Eval()
189 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kShape, &shape)); in Eval()
Dlsh_projection.cc77 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 0, &hash)); in Resize()
83 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 1, &input)); in Resize()
89 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 2, &weight)); in Resize()
182 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 0, &hash)); in Eval()
184 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 1, &input)); in Eval()
Dbasic_rnn.cc64 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Prepare()
67 context, GetInputSafe(context, node, kWeightsTensor, &input_weights)); in Prepare()
71 GetInputSafe(context, node, kRecurrentWeightsTensor, &recurrent_weights)); in Prepare()
73 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kBiasTensor, &bias)); in Prepare()
76 context, GetInputSafe(context, node, kHiddenStateTensor, &hidden_state)); in Prepare()
285 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Eval()
288 context, GetInputSafe(context, node, kWeightsTensor, &input_weights)); in Eval()
292 GetInputSafe(context, node, kRecurrentWeightsTensor, &recurrent_weights)); in Eval()
294 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kBiasTensor, &bias)); in Eval()
Dassign_variable.cc37 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputVariableId, in Prepare()
50 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputVariableId, in Eval()
54 context, GetInputSafe(context, node, kInputValue, &input_value_tensor)); in Eval()
Dslice.cc125 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Prepare()
127 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kBeginTensor, &begin)); in Prepare()
129 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kSizeTensor, &size)); in Prepare()
165 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Eval()
167 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kBeginTensor, &begin)); in Eval()
169 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kSizeTensor, &size)); in Eval()
Dsegment_sum.cc69 GetInputSafe(context, node, kInputDataTensor, &data)); in Prepare()
71 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputSegmentIdsTensor, in Prepare()
91 GetInputSafe(context, node, kInputDataTensor, &data)); in Eval()
93 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputSegmentIdsTensor, in Eval()
Dlogical.cc59 GetInputSafe(context, node, kInputTensor1, &input1)); in Prepare()
62 GetInputSafe(context, node, kInputTensor2, &input2)); in Prepare()
95 GetInputSafe(context, node, kInputTensor1, &input1)); in LogicalImpl()
98 GetInputSafe(context, node, kInputTensor2, &input2)); in LogicalImpl()
Dirfft2d.cc108 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in ResizeOutputandTemporaryTensors()
113 GetInputSafe(context, node, kFftLengthTensor, &fft_length)); in ResizeOutputandTemporaryTensors()
174 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Prepare()
186 GetInputSafe(context, node, kFftLengthTensor, &fft_length)); in Prepare()
299 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Irfft2dHelper()
303 GetInputSafe(context, node, kFftLengthTensor, &fft_length)); in Irfft2dHelper()
376 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Eval()
379 GetInputSafe(context, node, kFftLengthTensor, &fft_length)); in Eval()
Dtable.cc34 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Prepare()
36 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kTable, &table)); in Prepare()
100 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Eval()
102 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kTable, &table)); in Eval()
Dexpand_dims.cc78 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInput, &input)); in Prepare()
80 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kAxis, &axis)); in Prepare()
106 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInput, &input)); in Eval()
110 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kAxis, &axis)); in Eval()

12345