Home
last modified time | relevance | path

Searched refs:RequestScratchBufferInArena (Results 1 – 18 of 18) sorted by relevance

/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/
A Dmicro_interpreter.cc197 context_.RequestScratchBufferInArena = nullptr; in AllocateTensors()
204 context_.RequestScratchBufferInArena = RequestScratchBufferInArena; in AllocateTensors()
210 context_.RequestScratchBufferInArena = nullptr; in AllocateTensors()
318 TfLiteStatus MicroInterpreter::RequestScratchBufferInArena(TfLiteContext* ctx, in RequestScratchBufferInArena() function in tflite::MicroInterpreter
323 return interpreter->allocator_.RequestScratchBufferInArena( in RequestScratchBufferInArena()
A Dmicro_allocator.h186 TfLiteStatus RequestScratchBufferInArena(size_t bytes, int subgraph_idx,
A Dmicro_interpreter.h145 static TfLiteStatus RequestScratchBufferInArena(TfLiteContext* ctx,
A Dtest_helpers.cc659 TFLITE_DCHECK(context->RequestScratchBufferInArena == nullptr); in Init()
678 TF_LITE_ENSURE_STATUS(context->RequestScratchBufferInArena( in Prepare()
A Dmicro_allocator.cc681 TfLiteStatus MicroAllocator::RequestScratchBufferInArena(size_t bytes, in RequestScratchBufferInArena() function in tflite::MicroAllocator
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/
A Ddetection_postprocess.cc184 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 …]
A Dkernel_runner.cc48 context_.RequestScratchBufferInArena = RequestScratchBufferInArena; in KernelRunner()
113 TfLiteStatus KernelRunner::RequestScratchBufferInArena(TfLiteContext* context, in RequestScratchBufferInArena() function in tflite::micro::KernelRunner
A Dsvdf_common.cc438 TFLITE_DCHECK(context->RequestScratchBufferInArena != nullptr); in PrepareSvdf()
440 const TfLiteStatus scratch_status = context->RequestScratchBufferInArena( in PrepareSvdf()
446 context->RequestScratchBufferInArena( in PrepareSvdf()
459 TFLITE_DCHECK(context->RequestScratchBufferInArena != nullptr); in PrepareSvdf()
460 const TfLiteStatus scratch_status = context->RequestScratchBufferInArena( in PrepareSvdf()
A Dkernel_runner.h54 static TfLiteStatus RequestScratchBufferInArena(TfLiteContext* context,
A Dadd_n.cc82 TF_LITE_ENSURE_OK(context, context->RequestScratchBufferInArena( in CalculateOpData()
95 context, context->RequestScratchBufferInArena(context, scratch_size, in CalculateOpData()
A Dreduce.cc93 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()
A Dtranspose_conv.cc150 TFLITE_DCHECK(context->RequestScratchBufferInArena != nullptr); in Prepare()
151 TFLITE_DCHECK(context->RequestScratchBufferInArena( in Prepare()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/cmsis_nn/
A Dsvdf.cc392 TFLITE_DCHECK(context->RequestScratchBufferInArena != nullptr); in Prepare()
394 const TfLiteStatus scratch_status = context->RequestScratchBufferInArena( in Prepare()
400 context->RequestScratchBufferInArena( in Prepare()
413 TFLITE_DCHECK(context->RequestScratchBufferInArena != nullptr); in Prepare()
414 const TfLiteStatus scratch_status = context->RequestScratchBufferInArena( in Prepare()
A Dfully_connected.cc94 TF_LITE_ENSURE_STATUS(context->RequestScratchBufferInArena( in Prepare()
A Dconv.cc123 TF_LITE_ENSURE_STATUS(context->RequestScratchBufferInArena( in Prepare()
A Ddepthwise_conv.cc144 TF_LITE_ENSURE_STATUS(context->RequestScratchBufferInArena( in Prepare()
A Dpooling.cc314 TF_LITE_ENSURE_STATUS(context->RequestScratchBufferInArena( in AveragePrepare()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/c/
A Dcommon.h755 TfLiteStatus (*RequestScratchBufferInArena)(struct TfLiteContext* ctx, member

Completed in 27 milliseconds