| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | detection_postprocess.cc | 337 context->GetScratchBuffer(context, op_data->decoded_boxes_idx)); in DecodeCenterSizeBoxes()
422 context->GetScratchBuffer(context, op_data->decoded_boxes_idx)); in NonMaxSuppressionSingleClassHelper()
428 context->GetScratchBuffer(context, op_data->keep_indices_idx)); in NonMaxSuppressionSingleClassHelper()
430 context->GetScratchBuffer(context, op_data->keep_scores_idx)); in NonMaxSuppressionSingleClassHelper()
435 context->GetScratchBuffer(context, op_data->sorted_indices_idx)); in NonMaxSuppressionSingleClassHelper()
446 context->GetScratchBuffer(context, op_data->active_candidate_idx)); in NonMaxSuppressionSingleClassHelper()
513 context->GetScratchBuffer(context, op_data->score_buffer_idx)); in NonMaxSuppressionMultiClassRegularHelper()
515 context->GetScratchBuffer(context, op_data->buffer_idx)); in NonMaxSuppressionMultiClassRegularHelper()
517 reinterpret_cast<float*>(context->GetScratchBuffer( in NonMaxSuppressionMultiClassRegularHelper()
522 context->GetScratchBuffer(context, op_data->sorted_indices_idx)); in NonMaxSuppressionMultiClassRegularHelper()
[all …]
|
| D | kernel_runner.cc | 49 context_.GetScratchBuffer = GetScratchBuffer; in KernelRunner()
140 void* KernelRunner::GetScratchBuffer(TfLiteContext* context, int buffer_index) { in GetScratchBuffer() function in tflite::micro::KernelRunner
|
| D | svdf_common.cc | 49 TFLITE_DCHECK(context->GetScratchBuffer != nullptr); in EvalIntegerSvdfReference()
52 context->GetScratchBuffer(context, data.scratch_tensor_index)); in EvalIntegerSvdfReference()
54 context->GetScratchBuffer(context, data.scratch_output_tensor_index)); in EvalIntegerSvdfReference()
|
| D | reduce.cc | 192 context->GetScratchBuffer(context, op_data->temp_buffer_idx)); in EvalMean()
203 context->GetScratchBuffer(context, op_data->temp_buffer_idx)); in EvalMean()
228 context->GetScratchBuffer(context, op_data->temp_buffer_idx)); in EvalMean()
240 context->GetScratchBuffer(context, op_data->temp_buffer_idx)); in EvalMean()
273 context->GetScratchBuffer(context, op_data->temp_buffer_idx)); in EvalMax()
275 context->GetScratchBuffer(context, op_data->resolved_axis_idx)); in EvalMax()
|
| D | kernel_runner.h | 57 static void* GetScratchBuffer(TfLiteContext* context, int buffer_index);
|
| D | svdf.cc | 140 TFLITE_DCHECK(context->GetScratchBuffer != nullptr); in EvalFloatSVDF()
143 context->GetScratchBuffer(context, scratch_tensor_index)); in EvalFloatSVDF()
|
| D | transpose_conv.cc | 234 context->GetScratchBuffer(context, data.scratch_buffer_index)); in Eval()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ethos_u/ |
| D | ethosu.cc | 65 TFLITE_DCHECK(context->GetScratchBuffer != nullptr); in Eval()
76 context->GetScratchBuffer(context, data->base_addr_idx)); in Eval()
78 context->GetScratchBuffer(context, data->base_addr_size_idx)); in Eval()
|
| /external/tensorflow/tensorflow/lite/micro/ |
| D | micro_interpreter.cc | 66 void* ContextHelper::GetScratchBuffer(TfLiteContext* ctx, int buffer_idx) { in GetScratchBuffer() function in tflite::internal::ContextHelper
201 context_.GetScratchBuffer = nullptr; in AllocateTensors()
245 context_.GetScratchBuffer = context_helper_.GetScratchBuffer; in AllocateTensors()
|
| D | micro_interpreter.h | 53 static void* GetScratchBuffer(TfLiteContext* ctx, int buffer_idx);
|
| D | test_helpers.cc | 658 TFLITE_DCHECK(context->GetScratchBuffer == nullptr); in Init()
700 context->GetScratchBuffer(context, data->sorting_buffer)); in Invoke()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/ |
| D | svdf.cc | 156 TFLITE_DCHECK(context->GetScratchBuffer != nullptr); in EvalFloatSVDF()
159 context->GetScratchBuffer(context, scratch_tensor_index)); in EvalFloatSVDF()
257 TFLITE_DCHECK(context->GetScratchBuffer != nullptr); in EvalIntegerSVDF()
261 context->GetScratchBuffer(context, data.scratch_tensor_index)); in EvalIntegerSVDF()
265 context->GetScratchBuffer(context, data.scratch_output_tensor_index)); in EvalIntegerSVDF()
|
| D | pooling.cc | 156 ctx.buf = context->GetScratchBuffer(context, data.buffer_idx); in AverageEvalQuantized()
246 ctx.buf = context->GetScratchBuffer(context, data.buffer_idx); in MaxEvalInt8()
|
| D | fully_connected.cc | 157 ctx.buf = context->GetScratchBuffer(context, data.buffer_idx); in EvalQuantizedInt8()
|
| D | conv.cc | 209 ctx.buf = context->GetScratchBuffer(context, data.buffer_idx); in EvalQuantizedPerChannel()
|
| D | depthwise_conv.cc | 315 ctx.buf = context->GetScratchBuffer(context, data->buffer_idx); in EvalQuantizedPerChannel()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/xtensa/ |
| D | svdf.cc | 70 TFLITE_DCHECK(context->GetScratchBuffer != nullptr); in EvalIntegerSvdfHifimini()
73 context->GetScratchBuffer(context, data.scratch_tensor_index)); in EvalIntegerSvdfHifimini()
76 context->GetScratchBuffer(context, data.scratch_output_tensor_index)); in EvalIntegerSvdfHifimini()
265 TFLITE_DCHECK(context->GetScratchBuffer != nullptr); in EvalIntegerSvdfHifi4()
|
| D | softmax.cc | 225 context->GetScratchBuffer(context, op_data->scratch_tensor_index)); in EvalHifi4()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | common.h | 723 void* (*GetScratchBuffer)(struct TfLiteContext* ctx, int buffer_idx); member
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval_test.cc | 652 TfLiteTensor* GetScratchBuffer() { in GetScratchBuffer() function in tflite::__anonb312a86f0111::HybridLstmParam
934 /*output_offset=*/0, one_parameter.GetScratchBuffer(), in TestOneHybridAsymmLSTM()
|