Home
last modified time | relevance | path

Searched refs:TF_LITE_ENSURE_OK (Results 1 – 25 of 149) sorted by relevance

123456

/external/tensorflow/tensorflow/lite/kernels/
Dbidirectional_sequence_lstm.cc197 TF_LITE_ENSURE_OK(context, in CheckLstmTensorDimensionsAndTypes()
218 TF_LITE_ENSURE_OK(context, in CheckLstmTensorDimensionsAndTypes()
228 TF_LITE_ENSURE_OK(context, in CheckLstmTensorDimensionsAndTypes()
250 TF_LITE_ENSURE_OK( in CheckLstmTensorDimensionsAndTypes()
262 TF_LITE_ENSURE_OK( in CheckLstmTensorDimensionsAndTypes()
331 TF_LITE_ENSURE_OK( in CheckLstmTensorDimensionsAndTypes()
339 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, cell_gate_bias_tensor, in CheckLstmTensorDimensionsAndTypes()
346 TF_LITE_ENSURE_OK( in CheckLstmTensorDimensionsAndTypes()
386 TF_LITE_ENSURE_OK( in CheckInputTensorDimensions()
400 TF_LITE_ENSURE_OK( in CheckInputTensorDimensions()
[all …]
Dbasic_rnn.cc64 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Prepare()
66 TF_LITE_ENSURE_OK( in Prepare()
69 TF_LITE_ENSURE_OK( in Prepare()
73 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kBiasTensor, &bias)); in Prepare()
75 TF_LITE_ENSURE_OK( in Prepare()
97 TF_LITE_ENSURE_OK(context, in Prepare()
104 TF_LITE_ENSURE_OK(context, in Prepare()
118 TF_LITE_ENSURE_OK(context, GetTemporarySafe(context, node, /*index=*/0, in Prepare()
124 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, input_quantized, in Prepare()
129 TF_LITE_ENSURE_OK(context, GetTemporarySafe(context, node, /*index=*/1, in Prepare()
[all …]
Dnon_max_suppression.cc83 TF_LITE_ENSURE_OK( in Prepare()
90 TF_LITE_ENSURE_OK( in Prepare()
98 TF_LITE_ENSURE_OK(context, in Prepare()
112 TF_LITE_ENSURE_OK(context, in Prepare()
118 TF_LITE_ENSURE_OK(context, in Prepare()
126 TF_LITE_ENSURE_OK( in Prepare()
133 TF_LITE_ENSURE_OK( in Prepare()
139 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, in Prepare()
144 TF_LITE_ENSURE_OK( in Prepare()
161 TF_LITE_ENSURE_OK( in Prepare()
[all …]
Dsvdf.cc86 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Prepare()
88 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kWeightsFeatureTensor, in Prepare()
91 TF_LITE_ENSURE_OK( in Prepare()
115 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kStateTensor, &state)); in Prepare()
117 TF_LITE_ENSURE_OK(context, in Prepare()
130 TF_LITE_ENSURE_OK(context, in Prepare()
153 TF_LITE_ENSURE_OK( in Prepare()
164 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, scratch_tensor, in Prepare()
173 TF_LITE_ENSURE_OK(context, GetTemporarySafe(context, node, /*index=*/1, in Prepare()
179 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, input_quantized, in Prepare()
[all …]
Dunidirectional_sequence_lstm.cc61 TF_LITE_ENSURE_OK( in PopulateQuantizedLstmParams8x8_16()
88 TF_LITE_ENSURE_OK( in PopulateQuantizedLstmParams8x8_16()
94 TF_LITE_ENSURE_OK( in PopulateQuantizedLstmParams8x8_16()
99 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, in PopulateQuantizedLstmParams8x8_16()
103 TF_LITE_ENSURE_OK( in PopulateQuantizedLstmParams8x8_16()
111 TF_LITE_ENSURE_OK( in PopulateQuantizedLstmParams8x8_16()
116 TF_LITE_ENSURE_OK( in PopulateQuantizedLstmParams8x8_16()
121 TF_LITE_ENSURE_OK( in PopulateQuantizedLstmParams8x8_16()
161 TF_LITE_ENSURE_OK(context, in PopulateQuantizedLstmParams8x8_16()
176 TF_LITE_ENSURE_OK(context, GetIntermediatesSafe(context, node, 4, &hidden)); in PopulateQuantizedLstmParams8x8_16()
[all …]
Dwhile.cc62 TF_LITE_ENSURE_OK( in CopyTensorsShapeAndType()
154 TF_LITE_ENSURE_OK( in Prepare()
158 TF_LITE_ENSURE_OK(context, cond_subgraph->AllocateTensors()); in Prepare()
171 TF_LITE_ENSURE_OK( in Prepare()
175 TF_LITE_ENSURE_OK(context, body_subgraph->AllocateTensors()); in Prepare()
200 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, i, &output)); in Prepare()
207 TF_LITE_ENSURE_OK(context, in Prepare()
260 TF_LITE_ENSURE_OK( in Eval()
264 TF_LITE_ENSURE_OK(context, cond_subgraph->AllocateTensors()); in Eval()
266 TF_LITE_ENSURE_OK( in Eval()
[all …]
Dunidirectional_sequence_rnn.cc65 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Prepare()
67 TF_LITE_ENSURE_OK( in Prepare()
70 TF_LITE_ENSURE_OK( in Prepare()
74 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kBiasTensor, &bias)); in Prepare()
76 TF_LITE_ENSURE_OK( in Prepare()
103 TF_LITE_ENSURE_OK(context, in Prepare()
111 TF_LITE_ENSURE_OK(context, in Prepare()
125 TF_LITE_ENSURE_OK(context, GetTemporarySafe(context, node, /*index=*/0, in Prepare()
131 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, input_quantized, in Prepare()
136 TF_LITE_ENSURE_OK(context, GetTemporarySafe(context, node, /*index=*/1, in Prepare()
[all …]
Dlstm.cc153 TF_LITE_ENSURE_OK( in PopulateQuantizedLstmParams8x8_16()
179 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in PopulateQuantizedLstmParams8x8_16()
184 TF_LITE_ENSURE_OK(context, in PopulateQuantizedLstmParams8x8_16()
188 TF_LITE_ENSURE_OK(context, in PopulateQuantizedLstmParams8x8_16()
192 TF_LITE_ENSURE_OK(context, in PopulateQuantizedLstmParams8x8_16()
199 TF_LITE_ENSURE_OK(context, in PopulateQuantizedLstmParams8x8_16()
203 TF_LITE_ENSURE_OK(context, in PopulateQuantizedLstmParams8x8_16()
207 TF_LITE_ENSURE_OK(context, in PopulateQuantizedLstmParams8x8_16()
246 TF_LITE_ENSURE_OK(context, in PopulateQuantizedLstmParams8x8_16()
261 TF_LITE_ENSURE_OK(context, GetIntermediatesSafe(context, node, 4, &hidden)); in PopulateQuantizedLstmParams8x8_16()
[all …]
Dbidirectional_sequence_rnn.cc101 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Prepare()
103 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kFwWeightsTensor, in Prepare()
106 TF_LITE_ENSURE_OK(context, in Prepare()
110 TF_LITE_ENSURE_OK(context, in Prepare()
113 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kFwHiddenStateTensor, in Prepare()
116 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kBwWeightsTensor, in Prepare()
119 TF_LITE_ENSURE_OK(context, in Prepare()
123 TF_LITE_ENSURE_OK(context, in Prepare()
126 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kBwHiddenStateTensor, in Prepare()
203 TF_LITE_ENSURE_OK(context, GetTemporarySafe(context, node, kInputQuantized, in Prepare()
[all …]
Dtopk_v2.cc39 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTopK, &top_k)); in ResizeOutput()
47 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in ResizeOutput()
65 TF_LITE_ENSURE_OK( in ResizeOutput()
68 TF_LITE_ENSURE_OK( in ResizeOutput()
83 TF_LITE_ENSURE_OK(context, resize_tensor(output_indexes, output_indexes_shape, in ResizeOutput()
85 TF_LITE_ENSURE_OK(context, in ResizeOutput()
205 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Prepare()
207 TF_LITE_ENSURE_OK( in Prepare()
212 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTopK, &top_k)); in Prepare()
217 TF_LITE_ENSURE_OK(context, ResizeOutput(context, node)); in Prepare()
[all …]
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()
80 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, 0, &output)); in Prepare()
84 TF_LITE_ENSURE_OK(context, in Prepare()
95 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInput, &input)); in Eval()
97 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, 0, &output)); in Eval()
99 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kAxis, &axis)); in Eval()
102 TF_LITE_ENSURE_OK(context, in Eval()
104 TF_LITE_ENSURE_OK(context, in Eval()
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()
117 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, 0, &output)); in Prepare()
150 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, 0, &output)); in Eval()
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()
[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()
78 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, 1, &hits)); in Prepare()
84 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, 0, &output)); in Prepare()
104 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, 0, &output)); in Eval()
106 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, 1, &hits)); in Eval()
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()
Dsparse_to_dense.cc147 TF_LITE_ENSURE_OK(context, in Prepare()
150 TF_LITE_ENSURE_OK( in Prepare()
153 TF_LITE_ENSURE_OK(context, in Prepare()
156 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kDefaultValueTensor, in Prepare()
184 TF_LITE_ENSURE_OK( in Prepare()
188 TF_LITE_ENSURE_OK(context, in Prepare()
203 TF_LITE_ENSURE_OK(context, in SparseToDenseImpl()
206 TF_LITE_ENSURE_OK( in SparseToDenseImpl()
209 TF_LITE_ENSURE_OK(context, in SparseToDenseImpl()
212 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kDefaultValueTensor, in SparseToDenseImpl()
[all …]
Drfft2d.cc77 TF_LITE_ENSURE_OK( in InitTemporaryTensors()
87 TF_LITE_ENSURE_OK(context, in InitTemporaryTensors()
108 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in ResizeOutputandTemporaryTensors()
112 TF_LITE_ENSURE_OK(context, in ResizeOutputandTemporaryTensors()
127 TF_LITE_ENSURE_OK(context, in ResizeOutputandTemporaryTensors()
136 TF_LITE_ENSURE_OK( in ResizeOutputandTemporaryTensors()
147 TF_LITE_ENSURE_OK(context, in ResizeOutputandTemporaryTensors()
174 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Prepare()
185 TF_LITE_ENSURE_OK(context, in Prepare()
203 TF_LITE_ENSURE_OK(context, in Prepare()
[all …]
Drange.cc60 TF_LITE_ENSURE_OK(context, in ResizeOutput()
67 TF_LITE_ENSURE_OK(context, GetSize(context, *GetTensorData<float>(start), in ResizeOutput()
87 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()
110 TF_LITE_ENSURE_OK(context, 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()
146 TF_LITE_ENSURE_OK(context, in Eval()
[all …]
Dcomparisons.cc45 TF_LITE_ENSURE_OK(context, in ComparisonPrepareCommon()
48 TF_LITE_ENSURE_OK(context, in ComparisonPrepareCommon()
51 TF_LITE_ENSURE_OK(context, in ComparisonPrepareCommon()
66 TF_LITE_ENSURE_OK(context, CalculateShapeForBroadcast( in ComparisonPrepareCommon()
155 TF_LITE_ENSURE_OK(context, in EqualEval()
158 TF_LITE_ENSURE_OK(context, in EqualEval()
161 TF_LITE_ENSURE_OK(context, in EqualEval()
205 TF_LITE_ENSURE_OK(context, in NotEqualEval()
208 TF_LITE_ENSURE_OK(context, in NotEqualEval()
211 TF_LITE_ENSURE_OK(context, in NotEqualEval()
[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()
103 TF_LITE_ENSURE_OK(context, in Prepare()
110 TF_LITE_ENSURE_OK( in Prepare()
143 TF_LITE_ENSURE_OK( in EvalScatterNd()
147 TF_LITE_ENSURE_OK(context, in EvalScatterNd()
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()
[all …]
Dif.cc57 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 0, &cond)); in Prepare()
89 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, i + 1, &input)); in Prepare()
99 TF_LITE_ENSURE_OK(context, subgraph->AllocateTensors()); in Prepare()
120 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, i, &output)); in Prepare()
129 TF_LITE_ENSURE_OK(context, in Prepare()
141 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, 0, &cond)); in Eval()
156 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, i + 1, &input)); in Eval()
170 TF_LITE_ENSURE_OK(context, active_branch_subgraph.Invoke()); in Eval()
179 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, i, &output)); in Eval()
189 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, i, &output)); in Eval()
[all …]
Dselect.cc65 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensorCondition, in SelectPrepare()
68 TF_LITE_ENSURE_OK(context, in SelectPrepare()
71 TF_LITE_ENSURE_OK(context, in SelectPrepare()
74 TF_LITE_ENSURE_OK(context, in SelectPrepare()
103 TF_LITE_ENSURE_OK(context, CalculateShapeForBroadcast( in SelectPrepare()
122 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensorCondition, in SelectEval()
125 TF_LITE_ENSURE_OK(context, in SelectEval()
128 TF_LITE_ENSURE_OK(context, in SelectEval()
131 TF_LITE_ENSURE_OK(context, in SelectEval()
Dpow.cc58 TF_LITE_ENSURE_OK(context, in Prepare()
61 TF_LITE_ENSURE_OK(context, in Prepare()
64 TF_LITE_ENSURE_OK(context, in Prepare()
81 TF_LITE_ENSURE_OK(context, CalculateShapeForBroadcast( in Prepare()
122 TF_LITE_ENSURE_OK(context, in Eval()
125 TF_LITE_ENSURE_OK(context, in Eval()
128 TF_LITE_ENSURE_OK(context, in Eval()
134 TF_LITE_ENSURE_OK(context, CheckValue(context, input2)); in Eval()
/external/tensorflow/tensorflow/lite/experimental/kernels/
Dunidirectional_sequence_gru.cc131 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInput, &input)); in Prepare()
139 TF_LITE_ENSURE_OK(context, in Prepare()
147 TF_LITE_ENSURE_OK(context, in Prepare()
155 TF_LITE_ENSURE_OK(context, in Prepare()
162 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kCandidateWeight, in Prepare()
171 TF_LITE_ENSURE_OK( in Prepare()
178 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, kOutput, &output)); in Prepare()
183 TF_LITE_ENSURE_OK(context, in Prepare()
188 TF_LITE_ENSURE_OK(context, in Prepare()
190 TF_LITE_ENSURE_OK( in Prepare()
[all …]
Dctc_beam_search_decoder.cc66 TF_LITE_ENSURE_OK(context, in Prepare()
74 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kSequenceLengthTensor, in Prepare()
85 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, i, &indices)); in Prepare()
88 TF_LITE_ENSURE_OK(context, in Prepare()
92 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, i + 2 * top_paths, in Prepare()
99 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, top_paths * 3, in Prepare()
140 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, p, &indices)); in StoreAllDecodedSequences()
141 TF_LITE_ENSURE_OK(context, Resize(context, {p_num, 2}, indices)); in StoreAllDecodedSequences()
144 TF_LITE_ENSURE_OK(context, in StoreAllDecodedSequences()
146 TF_LITE_ENSURE_OK(context, Resize(context, {p_num}, values)); in StoreAllDecodedSequences()
[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()
53 TF_LITE_ENSURE_OK(context, in Prepare()
58 TF_LITE_ENSURE_OK(context, ResizeOutput(context, dims, output)); in Prepare()
67 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kValueTensor, &value)); in Eval()
70 TF_LITE_ENSURE_OK(context, in Eval()
75 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kDimsTensor, &dims)); in Eval()
76 TF_LITE_ENSURE_OK(context, ResizeOutput(context, dims, output)); in Eval()
/external/tensorflow/tensorflow/lite/kernels/hashtable/
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()
51 TF_LITE_ENSURE_OK(context, in PrepareHashtableFind()
54 TF_LITE_ENSURE_OK( in PrepareHashtableFind()
67 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputResourceIdTensor, in EvalHashtableFind()
72 TF_LITE_ENSURE_OK(context, in EvalHashtableFind()
75 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kDefaultValueTensor, in EvalHashtableFind()
78 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, 0, &output_tensor)); in EvalHashtableFind()

123456