| /external/tensorflow/tensorflow/lite/micro/ |
| D | micro_interpreter.cc | 59 TfLiteStatus ContextHelper::RequestScratchBufferInArena(TfLiteContext* ctx, in RequestScratchBufferInArena() function in tflite::internal::ContextHelper
63 return helper->allocator_->RequestScratchBufferInArena(bytes, buffer_idx); in RequestScratchBufferInArena()
200 context_.RequestScratchBufferInArena = nullptr; in AllocateTensors()
223 context_.RequestScratchBufferInArena = in AllocateTensors()
224 context_helper_.RequestScratchBufferInArena; in AllocateTensors()
244 context_.RequestScratchBufferInArena = nullptr; in AllocateTensors()
|
| D | micro_allocator.h | 174 TfLiteStatus RequestScratchBufferInArena(size_t bytes, int* buffer_idx);
|
| D | micro_interpreter.h | 50 static TfLiteStatus RequestScratchBufferInArena(TfLiteContext* ctx,
|
| D | test_helpers.cc | 659 TFLITE_DCHECK(context->RequestScratchBufferInArena == nullptr); in Init()
678 TF_LITE_ENSURE_STATUS(context->RequestScratchBufferInArena( in Prepare()
|
| D | micro_allocator.cc | 679 TfLiteStatus MicroAllocator::RequestScratchBufferInArena(size_t bytes, in RequestScratchBufferInArena() function in tflite::MicroAllocator
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | detection_postprocess.cc | 184 context->RequestScratchBufferInArena(context, num_boxes, in Prepare()
186 context->RequestScratchBufferInArena(context, in Prepare()
189 context->RequestScratchBufferInArena( in Prepare()
196 context->RequestScratchBufferInArena(context, num_boxes * sizeof(float), in Prepare()
198 context->RequestScratchBufferInArena(context, num_boxes * sizeof(float), in Prepare()
200 context->RequestScratchBufferInArena( in Prepare()
203 context->RequestScratchBufferInArena( in Prepare()
206 context->RequestScratchBufferInArena(context, num_boxes * sizeof(int), in Prepare()
208 context->RequestScratchBufferInArena( in Prepare()
212 context->RequestScratchBufferInArena( in Prepare()
[all …]
|
| D | kernel_runner.cc | 48 context_.RequestScratchBufferInArena = RequestScratchBufferInArena; in KernelRunner()
113 TfLiteStatus KernelRunner::RequestScratchBufferInArena(TfLiteContext* context, in RequestScratchBufferInArena() function in tflite::micro::KernelRunner
|
| D | svdf.cc | 302 TFLITE_DCHECK(context->RequestScratchBufferInArena != nullptr); in Prepare()
304 const TfLiteStatus scratch_status = context->RequestScratchBufferInArena( in Prepare()
310 context->RequestScratchBufferInArena( in Prepare()
323 TFLITE_DCHECK(context->RequestScratchBufferInArena != nullptr); in Prepare()
324 const TfLiteStatus scratch_status = context->RequestScratchBufferInArena( in Prepare()
|
| D | kernel_runner.h | 54 static TfLiteStatus RequestScratchBufferInArena(TfLiteContext* context,
|
| D | reduce.cc | 93 context->RequestScratchBufferInArena(context, sizeof(int) * input->dims->size, in PrepareMax()
95 context->RequestScratchBufferInArena( in PrepareMax()
114 context->RequestScratchBufferInArena(context, output_size * sizeof(int32_t), in PrepareMeanOrSum()
|
| D | transpose_conv.cc | 150 TFLITE_DCHECK(context->RequestScratchBufferInArena != nullptr); in Prepare()
151 TFLITE_DCHECK(context->RequestScratchBufferInArena( in Prepare()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/ |
| D | svdf.cc | 388 TFLITE_DCHECK(context->RequestScratchBufferInArena != nullptr); in Prepare()
390 const TfLiteStatus scratch_status = context->RequestScratchBufferInArena( in Prepare()
396 context->RequestScratchBufferInArena( in Prepare()
409 TFLITE_DCHECK(context->RequestScratchBufferInArena != nullptr); in Prepare()
410 const TfLiteStatus scratch_status = context->RequestScratchBufferInArena( in Prepare()
|
| D | fully_connected.cc | 90 TF_LITE_ENSURE_STATUS(context->RequestScratchBufferInArena( in Prepare()
|
| D | conv.cc | 120 TF_LITE_ENSURE_STATUS(context->RequestScratchBufferInArena( in Prepare()
|
| D | pooling.cc | 307 TF_LITE_ENSURE_STATUS(context->RequestScratchBufferInArena( in AveragePrepare()
|
| D | depthwise_conv.cc | 202 TF_LITE_ENSURE_STATUS(context->RequestScratchBufferInArena( in Prepare()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ethos_u/ |
| D | ethosu.cc | 50 TF_LITE_ENSURE_STATUS(context->RequestScratchBufferInArena( in Prepare()
52 TF_LITE_ENSURE_STATUS(context->RequestScratchBufferInArena( in Prepare()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/xtensa/ |
| D | svdf.cc | 437 const TfLiteStatus scratch_status = context->RequestScratchBufferInArena( in Prepare()
442 context->RequestScratchBufferInArena( in Prepare()
|
| D | softmax.cc | 205 context, context->RequestScratchBufferInArena( in PrepareHifi4()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | common.h | 717 TfLiteStatus (*RequestScratchBufferInArena)(struct TfLiteContext* ctx, member
|