Home
last modified time | relevance | path

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

12345

/external/tensorflow/tensorflow/lite/kernels/
Dembedding_lookup_sparse.cc87 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 0, &ids)); in Prepare()
92 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 1, &indices)); in Prepare()
97 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 2, &shape)); in Prepare()
102 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 3, &weights)); in Prepare()
112 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 4, &value)); in Prepare()
152 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 0, &ids)); in Eval()
154 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 1, &indices)); in Eval()
156 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 2, &dense_shape)); in Eval()
158 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 3, &weights)); in Eval()
160 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 4, &value)); in Eval()
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 …]
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()
156 GetInputSafe(context, node, kInputTensor1, &input1)); in EqualEval()
159 GetInputSafe(context, node, kInputTensor2, &input2)); in EqualEval()
206 GetInputSafe(context, node, kInputTensor1, &input1)); in NotEqualEval()
209 GetInputSafe(context, node, kInputTensor2, &input2)); in NotEqualEval()
256 GetInputSafe(context, node, kInputTensor1, &input1)); in GreaterEval()
259 GetInputSafe(context, node, kInputTensor2, &input2)); in GreaterEval()
297 GetInputSafe(context, node, kInputTensor1, &input1)); in GreaterEqualEval()
300 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.cc89 context, GetInputSafe(context, node, lstm::full::kInputTensor, &input)); in PopulateQuantizedLstmParams8x8_16()
96 GetInputSafe(context, node, lstm::full::kInputToForgetWeightsTensor, in PopulateQuantizedLstmParams8x8_16()
99 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, in PopulateQuantizedLstmParams8x8_16()
105 GetInputSafe(context, node, lstm::full::kInputToOutputWeightsTensor, in PopulateQuantizedLstmParams8x8_16()
113 GetInputSafe(context, node, lstm::full::kRecurrentToForgetWeightsTensor, in PopulateQuantizedLstmParams8x8_16()
118 GetInputSafe(context, node, lstm::full::kRecurrentToCellWeightsTensor, in PopulateQuantizedLstmParams8x8_16()
123 GetInputSafe(context, node, lstm::full::kRecurrentToOutputWeightsTensor, in PopulateQuantizedLstmParams8x8_16()
432 GetInputSafe(context, node, lstm::full::kInputToForgetWeightsTensor, in CheckInputTensorDimensions()
439 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, in CheckInputTensorDimensions()
459 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()
122 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensorCondition, in SelectEval()
126 GetInputSafe(context, node, kInputTensorX, &input_x)); in SelectEval()
129 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.cc179 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in PopulateQuantizedLstmParams8x8_16()
185 GetInputSafe(context, node, kInputToForgetWeightsTensor, in PopulateQuantizedLstmParams8x8_16()
189 GetInputSafe(context, node, kInputToCellWeightsTensor, in PopulateQuantizedLstmParams8x8_16()
193 GetInputSafe(context, node, kInputToOutputWeightsTensor, in PopulateQuantizedLstmParams8x8_16()
200 GetInputSafe(context, node, kRecurrentToForgetWeightsTensor, in PopulateQuantizedLstmParams8x8_16()
204 GetInputSafe(context, node, kRecurrentToCellWeightsTensor, in PopulateQuantizedLstmParams8x8_16()
208 GetInputSafe(context, node, kRecurrentToOutputWeightsTensor, in PopulateQuantizedLstmParams8x8_16()
468 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in PopulateQuantizedLstmParams8x8_8()
473 GetInputSafe(context, node, kInputToForgetWeightsTensor, in PopulateQuantizedLstmParams8x8_8()
477 GetInputSafe(context, node, kInputToCellWeightsTensor, in PopulateQuantizedLstmParams8x8_8()
[all …]
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()
172 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kIndices, &indices)); in Eval()
174 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kUpdates, &updates)); in Eval()
176 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kShape, &shape)); 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()
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()
88 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 2, &weight)); in Resize()
181 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 0, &hash)); in Eval()
183 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 1, &input)); in Eval()
Dassign_variable.cc44 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputVariableId, in Prepare()
56 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputVariableId, in Eval()
60 context, GetInputSafe(context, node, kInputValue, &input_value_tensor)); in Eval()
Dexpand_dims.cc76 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInput, &input)); in Prepare()
78 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kAxis, &axis)); in Prepare()
95 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInput, &input)); in Eval()
99 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kAxis, &axis)); 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()
284 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Eval()
287 context, GetInputSafe(context, node, kWeightsTensor, &input_weights)); in Eval()
291 GetInputSafe(context, node, kRecurrentWeightsTensor, &recurrent_weights)); in Eval()
293 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kBiasTensor, &bias)); 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()
Dslice.cc117 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Prepare()
119 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kBeginTensor, &begin)); in Prepare()
121 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kSizeTensor, &size)); in Prepare()
151 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Eval()
153 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kBeginTensor, &begin)); in Eval()
155 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kSizeTensor, &size)); 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()
/external/tensorflow/tensorflow/lite/kernels/hashtable/
Dhashtable_import.cc37 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputResourceIdTensor, in PrepareHashtableImport()
46 GetInputSafe(context, node, kKeyTensor, &key_tensor)); in PrepareHashtableImport()
49 GetInputSafe(context, node, kValueTensor, &value_tensor)); in PrepareHashtableImport()
62 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputResourceIdTensor, in EvalHashtableImport()
68 GetInputSafe(context, node, kKeyTensor, &key_tensor)); in EvalHashtableImport()
71 GetInputSafe(context, node, kValueTensor, &value_tensor)); in EvalHashtableImport()
Dhashtable_find.cc38 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputResourceIdTensor, in PrepareHashtableFind()
47 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kDefaultValueTensor, in PrepareHashtableFind()
52 GetInputSafe(context, node, kKeyTensor, &key_tensor)); in PrepareHashtableFind()
67 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputResourceIdTensor, in EvalHashtableFind()
73 GetInputSafe(context, node, kKeyTensor, &key_tensor)); in EvalHashtableFind()
75 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kDefaultValueTensor, in EvalHashtableFind()
/external/tensorflow/tensorflow/lite/experimental/kernels/
Dunidirectional_sequence_gru.cc131 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInput, &input)); in Prepare()
140 GetInputSafe(context, node, kInputState, &input_state)); in Prepare()
148 GetInputSafe(context, node, kGateWeight, &gate_weight)); in Prepare()
156 GetInputSafe(context, node, kGateBias, &gate_bias)); in Prepare()
162 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kCandidateWeight, in Prepare()
172 context, GetInputSafe(context, node, kCandidateBias, &candidate_bias)); in Prepare()
227 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInput, &input)); in Eval()
230 GetInputSafe(context, node, kInputState, &input_state)); in Eval()
233 GetInputSafe(context, node, kGateWeight, &gate_weight)); in Eval()
236 GetInputSafe(context, node, kGateBias, &gate_bias)); in Eval()
[all …]
/external/tensorflow/tensorflow/lite/micro/kernels/
Dfill.cc38 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kDimsTensor, &dims)); in Prepare()
40 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kValueTensor, &value)); in Prepare()
67 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kValueTensor, &value)); in Eval()
75 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kDimsTensor, &dims)); in Eval()
Dfloor_div.cc45 GetInputSafe(context, node, kInputTensor1, &input1)); in Prepare()
48 GetInputSafe(context, node, kInputTensor2, &input2)); in Prepare()
102 GetInputSafe(context, node, kInputTensor1, &input1)); in Eval()
105 GetInputSafe(context, node, kInputTensor2, &input2)); in Eval()
Dfloor_mod.cc50 GetInputSafe(context, node, kInputTensor1, &input1)); in Prepare()
53 GetInputSafe(context, node, kInputTensor2, &input2)); in Prepare()
106 GetInputSafe(context, node, kInputTensor1, &input1)); in Eval()
109 GetInputSafe(context, node, kInputTensor2, &input2)); in Eval()

12345