| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/ |
| A D | detection_postprocess.cc | 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()
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()
535 context->GetScratchBuffer(context, op_data->selected_idx)); in NonMaxSuppressionMultiClassRegularHelper()
647 context->GetScratchBuffer(context, op_data->score_buffer_idx)); in NonMaxSuppressionMultiClassFastHelper()
649 context->GetScratchBuffer(context, op_data->buffer_idx)); in NonMaxSuppressionMultiClassFastHelper()
663 context->GetScratchBuffer(context, op_data->selected_idx)); in NonMaxSuppressionMultiClassFastHelper()
[all …]
|
| A 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
|
| A 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()
|
| A D | svdf_common.cc | 68 TFLITE_DCHECK(context->GetScratchBuffer != nullptr); in EvalIntegerSvdfReference()
71 context->GetScratchBuffer(context, data.scratch_tensor_index)); in EvalIntegerSvdfReference()
73 context->GetScratchBuffer(context, data.scratch_output_tensor_index)); in EvalIntegerSvdfReference()
280 TFLITE_DCHECK(context->GetScratchBuffer != nullptr); in EvalFloatSvdfReference()
283 context->GetScratchBuffer(context, scratch_tensor_index)); in EvalFloatSvdfReference()
|
| A D | kernel_runner.h | 57 static void* GetScratchBuffer(TfLiteContext* context, int buffer_index);
|
| A D | add_n.cc | 138 void* scratch_buffer = context->GetScratchBuffer(context, scratch_index); in CopyInputsToScratchBuffer()
|
| A D | transpose_conv.cc | 234 context->GetScratchBuffer(context, data.scratch_buffer_index)); in Eval()
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/ |
| A D | micro_interpreter.cc | 198 context_.GetScratchBuffer = nullptr; in AllocateTensors()
211 context_.GetScratchBuffer = GetScratchBuffer; in AllocateTensors()
327 void* MicroInterpreter::GetScratchBuffer(TfLiteContext* ctx, int buffer_idx) { in GetScratchBuffer() function in tflite::MicroInterpreter
|
| A D | micro_interpreter.h | 148 static void* GetScratchBuffer(TfLiteContext* ctx, int buffer_idx);
|
| A D | test_helpers.cc | 658 TFLITE_DCHECK(context->GetScratchBuffer == nullptr); in Init()
700 context->GetScratchBuffer(context, data->sorting_buffer)); in Invoke()
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/cmsis_nn/ |
| A 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()
|
| A D | pooling.cc | 161 ctx.buf = context->GetScratchBuffer(context, data.buffer_idx); in AverageEvalQuantized()
251 ctx.buf = context->GetScratchBuffer(context, data.buffer_idx); in MaxEvalInt8()
|
| A D | fully_connected.cc | 158 ctx.buf = context->GetScratchBuffer(context, data.buffer_idx); in EvalQuantizedInt8()
|
| A D | conv.cc | 212 ctx.buf = context->GetScratchBuffer(context, data.buffer_idx); in EvalQuantizedPerChannel()
|
| A D | depthwise_conv.cc | 226 ctx.buf = context->GetScratchBuffer(context, data.buffer_idx); in EvalQuantizedPerChannel()
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/c/ |
| A D | common.h | 761 void* (*GetScratchBuffer)(struct TfLiteContext* ctx, int buffer_idx); member
|