/external/tensorflow/tensorflow/lite/kernels/ |
D | while.cc | 132 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 …]
|
D | bidirectional_sequence_lstm.cc | 352 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 …]
|
D | reduce.cc | 172 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 …]
|
D | expand_dims.cc | 81 TF_LITE_ENSURE_OK(context, in Prepare() 96 TF_LITE_ENSURE_OK(context, in Eval() 98 TF_LITE_ENSURE_OK(context, in Eval()
|
D | lstm.cc | 358 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 …]
|
D | unidirectional_sequence_lstm.cc | 284 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()
|
D | if.cc | 91 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()
|
D | range.cc | 54 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()
|
D | topk_v2.cc | 72 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()
|
D | basic_rnn.cc | 95 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()
|
D | svdf.cc | 175 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()
|
D | fill.cc | 89 TF_LITE_ENSURE_OK(context, ResizeOutput(context, dims, output)); in Prepare() 103 TF_LITE_ENSURE_OK(context, ResizeOutput(context, dims, output)); in Eval()
|
D | sparse_to_dense.cc | 177 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()
|
D | unidirectional_sequence_rnn.cc | 96 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()
|
D | bidirectional_sequence_rnn.cc | 194 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()
|
D | pow.cc | 70 TF_LITE_ENSURE_OK(context, CalculateShapeForBroadcast( in Prepare() 117 TF_LITE_ENSURE_OK(context, CheckValue(context, input2)); in Eval()
|
D | reshape.cc | 121 TF_LITE_ENSURE_OK( in Prepare() 138 TF_LITE_ENSURE_OK( in Eval()
|
D | tile.cc | 156 TF_LITE_ENSURE_OK(context, ResizeOutput(context, node)); in Prepare() 169 TF_LITE_ENSURE_OK(context, ResizeOutput(context, node)); in Eval()
|
D | sub.cc | 208 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()
|
D | transpose_conv.cc | 189 TF_LITE_ENSURE_OK(context, in Eval() 193 TF_LITE_ENSURE_OK(context, ResizeIm2ColTensor(context, output_shape, in Eval()
|
/external/tensorflow/tensorflow/lite/core/ |
D | subgraph.cc | 374 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/ |
D | token_encoder.cc | 57 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()
|
D | text_encoder.cc | 138 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()
|
D | dist_diversification.cc | 95 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/ |
D | ctc_beam_search_decoder.cc | 130 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()
|