Home
last modified time | relevance | path

Searched refs:AllocatePersistentBuffer (Results 1 – 25 of 45) sorted by relevance

12

/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/
A Dquantize.cc29 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in Init()
30 return context->AllocatePersistentBuffer(context, in Init()
A Dsoftmax_common.cc88 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in SoftmaxInit()
89 return context->AllocatePersistentBuffer(context, sizeof(SoftmaxParams)); in SoftmaxInit()
105 void* raw_exp_lut = context->AllocatePersistentBuffer( in SoftmaxPrepare()
109 void* one_over_one_plus_x_lut = context->AllocatePersistentBuffer( in SoftmaxPrepare()
A Dkernel_runner.cc47 context_.AllocatePersistentBuffer = AllocatePersistentBuffer; in KernelRunner()
104 void* KernelRunner::AllocatePersistentBuffer(TfLiteContext* context, in AllocatePersistentBuffer() function in tflite::micro::KernelRunner
A Dconcatenation.cc129 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in Init()
130 return context->AllocatePersistentBuffer(context, sizeof(OpData)); in Init()
197 reinterpret_cast<float*>(context->AllocatePersistentBuffer( in Prepare()
201 reinterpret_cast<int32_t*>(context->AllocatePersistentBuffer( in Prepare()
A Dtranspose_conv.cc114 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in Init()
115 return context->AllocatePersistentBuffer(context, sizeof(OpData)); in Init()
142 static_cast<int32_t*>(context->AllocatePersistentBuffer( in Prepare()
145 static_cast<int32_t*>(context->AllocatePersistentBuffer( in Prepare()
A Dactivations.cc133 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in ReluInit()
134 return context->AllocatePersistentBuffer(context, sizeof(ReluOpData)); in ReluInit()
196 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in Relu6Init()
197 return context->AllocatePersistentBuffer(context, sizeof(Relu6OpData)); in Relu6Init()
A Dspace_to_batch_nd.cc42 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in Init()
43 return context->AllocatePersistentBuffer(context, sizeof(SpaceToBatchParams)); in Init()
A Dhard_swish.cc38 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in HardSwishInit()
39 return context->AllocatePersistentBuffer(context, sizeof(HardSwishParams)); in HardSwishInit()
A Ddequantize.cc42 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in Init()
43 return context->AllocatePersistentBuffer(context, sizeof(OpData)); in Init()
A Dl2norm.cc77 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in Init()
78 return context->AllocatePersistentBuffer(context, in Init()
A Dlogistic.cc73 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in LogisticInit()
74 return context->AllocatePersistentBuffer(context, sizeof(OpData)); in LogisticInit()
A Dcircular_buffer.cc73 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in Init()
75 context->AllocatePersistentBuffer(context, sizeof(OpData))); in Init()
A Delu.cc104 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in EluInit()
105 return context->AllocatePersistentBuffer(context, sizeof(OpData)); in EluInit()
A Dleaky_relu.cc98 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in LeakyReluInit()
99 return context->AllocatePersistentBuffer(context, sizeof(LeakyReluOpData)); in LeakyReluInit()
A Dtanh.cc45 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in TanhInit()
46 return context->AllocatePersistentBuffer(context, sizeof(OpData)); in TanhInit()
A Dkernel_runner.h53 static void* AllocatePersistentBuffer(TfLiteContext* context, size_t bytes);
A Dprelu.cc89 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in PreluInit()
90 return context->AllocatePersistentBuffer(context, sizeof(PreluParams)); in PreluInit()
A Dstrided_slice.cc121 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in Init()
122 return context->AllocatePersistentBuffer(context, sizeof(StridedSliceParams)); in Init()
A Dkernel_util.cc64 context->AllocatePersistentBuffer(context, alloc_size)); in CreateWritableTensorDimsWithCopy()
A Dconv_common.cc151 static_cast<int32_t*>(context->AllocatePersistentBuffer( in ConvPrepare()
154 static_cast<int32_t*>(context->AllocatePersistentBuffer( in ConvPrepare()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/
A Dmicro_interpreter.cc196 context_.AllocatePersistentBuffer = AllocatePersistentBuffer; in AllocateTensors()
209 context_.AllocatePersistentBuffer = nullptr; in AllocateTensors()
220 reinterpret_cast<TfLiteTensor**>(allocator_.AllocatePersistentBuffer( in AllocateTensors()
244 reinterpret_cast<TfLiteTensor**>(allocator_.AllocatePersistentBuffer( in AllocateTensors()
312 void* MicroInterpreter::AllocatePersistentBuffer(TfLiteContext* ctx, in AllocatePersistentBuffer() function in tflite::MicroInterpreter
315 ->allocator_.AllocatePersistentBuffer(bytes); in AllocatePersistentBuffer()
A Drecording_micro_allocator.cc108 void* RecordingMicroAllocator::AllocatePersistentBuffer(size_t bytes) { in AllocatePersistentBuffer() function in tflite::RecordingMicroAllocator
110 void* buffer = MicroAllocator::AllocatePersistentBuffer(bytes); in AllocatePersistentBuffer()
A Drecording_micro_allocator.h71 void* AllocatePersistentBuffer(size_t bytes) override;
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/cmsis_nn/
A Dconv.cc42 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in Init()
43 return context->AllocatePersistentBuffer(context, sizeof(OpData)); in Init()
93 static_cast<int32_t*>(context->AllocatePersistentBuffer( in Prepare()
96 static_cast<int32_t*>(context->AllocatePersistentBuffer( in Prepare()
A Ddepthwise_conv.cc43 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in Init()
44 return context->AllocatePersistentBuffer(context, sizeof(OpData)); in Init()
95 reinterpret_cast<int32_t*>(context->AllocatePersistentBuffer( in Prepare()
98 reinterpret_cast<int32_t*>(context->AllocatePersistentBuffer( in Prepare()

Completed in 24 milliseconds

12