Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/lite/kernels/
Dwhile.cc132 TF_LITE_ENSURE_OK( in Prepare()
136 TF_LITE_ENSURE_OK(context, cond_subgraph->AllocateTensors()); in Prepare()
150 TF_LITE_ENSURE_OK( in Prepare()
154 TF_LITE_ENSURE_OK(context, body_subgraph->AllocateTensors()); in Prepare()
186 TF_LITE_ENSURE_OK(context, in Prepare()
235 TF_LITE_ENSURE_OK( in Eval()
241 TF_LITE_ENSURE_OK(context, cond_subgraph->Invoke()); in Eval()
254 TF_LITE_ENSURE_OK(context, in Eval()
258 TF_LITE_ENSURE_OK(context, body_subgraph->AllocateTensors()); in Eval()
261 TF_LITE_ENSURE_OK( in Eval()
[all …]
Dbidirectional_sequence_lstm.cc352 TF_LITE_ENSURE_OK( in CheckInputTensorDimensions()
366 TF_LITE_ENSURE_OK( in CheckInputTensorDimensions()
441 TF_LITE_ENSURE_OK( in Prepare()
508 TF_LITE_ENSURE_OK(context, in Prepare()
545 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, fw_scratch_buffer, in Prepare()
550 TF_LITE_ENSURE_OK( in Prepare()
567 TF_LITE_ENSURE_OK( in Prepare()
602 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, bw_scratch_buffer, in Prepare()
615 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, input_quantized, in Prepare()
629 TF_LITE_ENSURE_OK( in Prepare()
[all …]
Dreduce.cc172 TF_LITE_ENSURE_OK(context, in InitializeTemporaries()
212 TF_LITE_ENSURE_OK(context, InitializeTemporaries(context, node, &op_context)); in PrepareSimple()
222 TF_LITE_ENSURE_OK(context, in PrepareSimple()
224 TF_LITE_ENSURE_OK(context, ResizeOutputTensor(context, &op_context)); in PrepareSimple()
236 TF_LITE_ENSURE_OK(context, PrepareSimple(context, node)); in PrepareMeanOrSum()
280 TF_LITE_ENSURE_OK(context, in EvalMean()
282 TF_LITE_ENSURE_OK(context, ResizeOutputTensor(context, &op_context)); in EvalMean()
283 TF_LITE_ENSURE_OK(context, ResizeTempSum(context, &op_context, temp_sum)); in EvalMean()
409 TF_LITE_ENSURE_OK(context, in EvalLogic()
411 TF_LITE_ENSURE_OK(context, ResizeOutputTensor(context, op_context)); in EvalLogic()
[all …]
Dexpand_dims.cc81 TF_LITE_ENSURE_OK(context, in Prepare()
96 TF_LITE_ENSURE_OK(context, in Eval()
98 TF_LITE_ENSURE_OK(context, in Eval()
Dlstm.cc358 TF_LITE_ENSURE_OK(context, in Prepare()
380 TF_LITE_ENSURE_OK(context, in Prepare()
414 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, scratch_buffer, in Prepare()
426 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, input_quantized, in Prepare()
438 TF_LITE_ENSURE_OK( in Prepare()
450 TF_LITE_ENSURE_OK(context, in Prepare()
468 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, scaling_factors, in Prepare()
480 TF_LITE_ENSURE_OK(context, in Prepare()
497 TF_LITE_ENSURE_OK(context, in Prepare()
707 TF_LITE_ENSURE_OK(context, context->ResizeTensor( in Prepare()
[all …]
Dunidirectional_sequence_lstm.cc284 TF_LITE_ENSURE_OK(context, CheckInputTensorDimensions(context, node, n_input, in Prepare()
304 TF_LITE_ENSURE_OK(context, in Prepare()
338 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, scratch_buffer, in Prepare()
352 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, input_quantized, in Prepare()
365 TF_LITE_ENSURE_OK( in Prepare()
378 TF_LITE_ENSURE_OK(context, in Prepare()
398 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, scaling_factors, in Prepare()
411 TF_LITE_ENSURE_OK(context, in Prepare()
429 TF_LITE_ENSURE_OK(context, in Prepare()
Dif.cc91 TF_LITE_ENSURE_OK(context, subgraph->AllocateTensors()); in Prepare()
120 TF_LITE_ENSURE_OK(context, in Prepare()
154 TF_LITE_ENSURE_OK(context, active_branch_subgraph.Invoke()); in Eval()
175 TF_LITE_ENSURE_OK(context, in Eval()
Drange.cc54 TF_LITE_ENSURE_OK(context, in ResizeOutput()
61 TF_LITE_ENSURE_OK(context, GetSize(context, *GetTensorData<float>(start), in ResizeOutput()
140 TF_LITE_ENSURE_OK(context, in Eval()
Dtopk_v2.cc72 TF_LITE_ENSURE_OK(context, resize_tensor(output_indexes, output_indexes_shape, in ResizeOutput()
74 TF_LITE_ENSURE_OK(context, in ResizeOutput()
175 TF_LITE_ENSURE_OK(context, ResizeOutput(context, node)); in Prepare()
189 TF_LITE_ENSURE_OK(context, ResizeOutput(context, node)); in Eval()
Dbasic_rnn.cc95 TF_LITE_ENSURE_OK(context, in Prepare()
114 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, input_quantized, in Prepare()
126 TF_LITE_ENSURE_OK(context, in Prepare()
138 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, scaling_factors, in Prepare()
Dsvdf.cc175 TF_LITE_ENSURE_OK(context, in Prepare()
199 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, scratch_tensor, in Prepare()
211 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, input_quantized, in Prepare()
224 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, scaling_factors, in Prepare()
238 TF_LITE_ENSURE_OK(context, in Prepare()
Dfill.cc89 TF_LITE_ENSURE_OK(context, ResizeOutput(context, dims, output)); in Prepare()
103 TF_LITE_ENSURE_OK(context, ResizeOutput(context, dims, output)); in Eval()
Dsparse_to_dense.cc177 TF_LITE_ENSURE_OK( in Prepare()
201 TF_LITE_ENSURE_OK(context, in SparseToDenseImpl()
209 TF_LITE_ENSURE_OK(context, GetIndicesVector<TI>(context, indices, num_indices, in SparseToDenseImpl()
Dunidirectional_sequence_rnn.cc96 TF_LITE_ENSURE_OK(context, in Prepare()
115 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, input_quantized, in Prepare()
127 TF_LITE_ENSURE_OK(context, in Prepare()
139 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, scaling_factors, in Prepare()
Dbidirectional_sequence_rnn.cc194 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, input_quantized, in Prepare()
208 TF_LITE_ENSURE_OK( in Prepare()
223 TF_LITE_ENSURE_OK( in Prepare()
239 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, scaling_factors, in Prepare()
253 TF_LITE_ENSURE_OK(context, in Prepare()
267 TF_LITE_ENSURE_OK( in Prepare()
275 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, bw_output, in Prepare()
Dpow.cc70 TF_LITE_ENSURE_OK(context, CalculateShapeForBroadcast( in Prepare()
117 TF_LITE_ENSURE_OK(context, CheckValue(context, input2)); in Eval()
Dreshape.cc121 TF_LITE_ENSURE_OK( in Prepare()
138 TF_LITE_ENSURE_OK( in Eval()
Dtile.cc156 TF_LITE_ENSURE_OK(context, ResizeOutput(context, node)); in Prepare()
169 TF_LITE_ENSURE_OK(context, ResizeOutput(context, node)); in Eval()
Dsub.cc208 TF_LITE_ENSURE_OK(context, CalculateShapeForBroadcast( in Prepare()
215 TF_LITE_ENSURE_OK(context, Prepare8BitSubOp(context, input1, input2, output, in Prepare()
218 TF_LITE_ENSURE_OK(context, PrepareInt16SubOp(context, input1, input2, in Prepare()
Dtranspose_conv.cc189 TF_LITE_ENSURE_OK(context, in Eval()
193 TF_LITE_ENSURE_OK(context, ResizeIm2ColTensor(context, output_shape, in Eval()
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.cc374 TF_LITE_ENSURE_OK(&context_, in SetInputs()
381 TF_LITE_ENSURE_OK( in SetOutputs()
388 TF_LITE_ENSURE_OK(&context_, CheckTensorIndices("variables", variables.data(), in SetVariables()
531 TF_LITE_ENSURE_OK(context_, CheckTensorIndices("node inputs", inputs.data(), in AddNodeWithParameters()
533 TF_LITE_ENSURE_OK( in AddNodeWithParameters()
671 TF_LITE_ENSURE_OK(context_, nnapi_delegate_->Invoke(this)); in Invoke()
836 TF_LITE_ENSURE_OK(context_, in SetTensorParametersReadOnly()
886 TF_LITE_ENSURE_OK(context_, in SetTensorParametersReadWrite()
1004 TF_LITE_ENSURE_OK(&context_, PrepareOpsStartingAt( in ModifyGraphWithDelegate()
1025 TF_LITE_ENSURE_OK(context_, status); in ModifyGraphWithDelegate()
[all …]
/external/libtextclassifier/utils/tflite/
Dtoken_encoder.cc57 TF_LITE_ENSURE_OK( in ResizeOutputTensors()
65 TF_LITE_ENSURE_OK( in ResizeOutputTensors()
86 TF_LITE_ENSURE_OK(context, in Prepare()
130 TF_LITE_ENSURE_OK( in Eval()
Dtext_encoder.cc138 TF_LITE_ENSURE_OK( in ResizeOutputTensors()
144 TF_LITE_ENSURE_OK( in ResizeOutputTensors()
152 TF_LITE_ENSURE_OK( in ResizeOutputTensors()
175 TF_LITE_ENSURE_OK(context, in Prepare()
228 TF_LITE_ENSURE_OK( in Eval()
Ddist_diversification.cc95 TF_LITE_ENSURE_OK(context, AllocateOutputIndexes(context, node)); in Prepare()
104 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, &output_length, in Prepare()
114 TF_LITE_ENSURE_OK(context, AllocateOutputIndexes(context, node)); in Eval()
/external/tensorflow/tensorflow/lite/experimental/kernels/
Dctc_beam_search_decoder.cc130 TF_LITE_ENSURE_OK(context, Resize(context, {p_num, 2}, indices)); in StoreAllDecodedSequences()
133 TF_LITE_ENSURE_OK(context, Resize(context, {p_num}, values)); in StoreAllDecodedSequences()
136 TF_LITE_ENSURE_OK(context, Resize(context, {2}, decoded_shape)); in StoreAllDecodedSequences()

123