Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/lite/micro/
Dmicro_interpreter.cc53 void* ContextHelper::AllocatePersistentBuffer(TfLiteContext* ctx, in AllocatePersistentBuffer() function in tflite::internal::ContextHelper
56 ->allocator_->AllocatePersistentBuffer(bytes); in AllocatePersistentBuffer()
199 context_.AllocatePersistentBuffer = context_helper_.AllocatePersistentBuffer; in AllocateTensors()
243 context_.AllocatePersistentBuffer = nullptr; in AllocateTensors()
256 reinterpret_cast<TfLiteTensor**>(allocator_.AllocatePersistentBuffer( in AllocateTensors()
280 reinterpret_cast<TfLiteTensor**>(allocator_.AllocatePersistentBuffer( in AllocateTensors()
Drecording_micro_allocator.cc108 void* RecordingMicroAllocator::AllocatePersistentBuffer(size_t bytes) { in AllocatePersistentBuffer() function in tflite::RecordingMicroAllocator
110 void* buffer = MicroAllocator::AllocatePersistentBuffer(bytes); in AllocatePersistentBuffer()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dsoftmax_common.cc92 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in SoftmaxInit()
93 return context->AllocatePersistentBuffer(context, sizeof(SoftmaxParams)); in SoftmaxInit()
109 void* raw_exp_lut = context->AllocatePersistentBuffer( in SoftmaxPrepare()
113 void* one_over_one_plus_x_lut = context->AllocatePersistentBuffer( in SoftmaxPrepare()
Dkernel_runner.cc47 context_.AllocatePersistentBuffer = AllocatePersistentBuffer; in KernelRunner()
104 void* KernelRunner::AllocatePersistentBuffer(TfLiteContext* context, in AllocatePersistentBuffer() function in tflite::micro::KernelRunner
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()
Dquantize.cc29 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in Init()
30 return context->AllocatePersistentBuffer(context, in Init()
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()
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()
Dconv.cc33 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in Init()
34 return context->AllocatePersistentBuffer(context, sizeof(OpDataConv)); in Init()
Dcircular_buffer.cc71 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in Init()
72 return context->AllocatePersistentBuffer(context, sizeof(OpData)); in Init()
Dhard_swish.cc38 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in HardSwishInit()
39 return context->AllocatePersistentBuffer(context, sizeof(HardSwishParams)); in HardSwishInit()
Dspace_to_batch_nd.cc42 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in Init()
43 return context->AllocatePersistentBuffer(context, sizeof(SpaceToBatchParams)); in Init()
Dlogistic.cc73 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in LogisticInit()
74 return context->AllocatePersistentBuffer(context, sizeof(OpData)); in LogisticInit()
Dl2norm.cc78 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in Init()
79 return context->AllocatePersistentBuffer(context, in Init()
Ddepthwise_conv.cc102 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in Init()
103 return context->AllocatePersistentBuffer(context, sizeof(OpData)); in Init()
132 reinterpret_cast<int32_t*>(context->AllocatePersistentBuffer( in Prepare()
135 reinterpret_cast<int32_t*>(context->AllocatePersistentBuffer( in Prepare()
Dtanh.cc45 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in TanhInit()
46 return context->AllocatePersistentBuffer(context, sizeof(OpData)); in TanhInit()
Dfully_connected.cc32 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in Init()
33 return context->AllocatePersistentBuffer(context, in Init()
Ddequantize.cc42 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in Init()
43 return context->AllocatePersistentBuffer(context, sizeof(OpData)); in Init()
/external/tensorflow/tensorflow/lite/micro/kernels/arc_mli/
Ddepthwise_conv.cc139 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in Init()
140 return context->AllocatePersistentBuffer(context, sizeof(OpData)); in Init()
167 reinterpret_cast<int32_t*>(context->AllocatePersistentBuffer( in Prepare()
170 reinterpret_cast<int32_t*>(context->AllocatePersistentBuffer( in Prepare()
205 context->AllocatePersistentBuffer(context, sizeof(mli_tensor))); in Prepare()
207 context->AllocatePersistentBuffer(context, sizeof(mli_tensor))); in Prepare()
209 context->AllocatePersistentBuffer(context, sizeof(mli_tensor))); in Prepare()
211 context->AllocatePersistentBuffer(context, sizeof(mli_tensor))); in Prepare()
213 context->AllocatePersistentBuffer(context, sizeof(mli_conv2d_cfg))); in Prepare()
Dconv.cc149 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in Init()
150 return context->AllocatePersistentBuffer(context, sizeof(OpData)); in Init()
175 reinterpret_cast<int32_t*>(context->AllocatePersistentBuffer( in Prepare()
178 reinterpret_cast<int32_t*>(context->AllocatePersistentBuffer( in Prepare()
213 context->AllocatePersistentBuffer(context, sizeof(mli_tensor))); in Prepare()
215 context->AllocatePersistentBuffer(context, sizeof(mli_tensor))); in Prepare()
217 context->AllocatePersistentBuffer(context, sizeof(mli_tensor))); in Prepare()
219 context->AllocatePersistentBuffer(context, sizeof(mli_tensor))); in Prepare()
221 context->AllocatePersistentBuffer(context, sizeof(mli_conv2d_cfg))); in Prepare()
Dfully_connected.cc104 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in Init()
105 return context->AllocatePersistentBuffer(context, sizeof(OpData)); in Init()
134 context->AllocatePersistentBuffer(context, sizeof(mli_tensor))); in Prepare()
136 context->AllocatePersistentBuffer(context, sizeof(mli_tensor))); in Prepare()
138 context->AllocatePersistentBuffer(context, sizeof(mli_tensor))); in Prepare()
140 context->AllocatePersistentBuffer(context, sizeof(mli_tensor))); in Prepare()
Dpooling.cc82 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in Init()
83 return context->AllocatePersistentBuffer(context, sizeof(OpData)); in Init()
113 context->AllocatePersistentBuffer(context, sizeof(mli_tensor))); in Prepare()
115 context->AllocatePersistentBuffer(context, sizeof(mli_tensor))); in Prepare()
117 context->AllocatePersistentBuffer(context, sizeof(mli_pool_cfg))); in Prepare()
/external/tensorflow/tensorflow/lite/micro/kernels/xtensa/
Dquantize.cc174 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in Init()
176 return context->AllocatePersistentBuffer(context, sizeof(OpData)); in Init()
178 return context->AllocatePersistentBuffer(context, in Init()
/external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/
Dconv.cc42 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in Init()
43 return context->AllocatePersistentBuffer(context, sizeof(OpData)); in Init()
90 static_cast<int32_t*>(context->AllocatePersistentBuffer( in Prepare()
93 static_cast<int32_t*>(context->AllocatePersistentBuffer( in Prepare()
/external/tensorflow/tensorflow/lite/micro/kernels/ethos_u/
Dethosu.cc34 TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr); in Init()
35 return context->AllocatePersistentBuffer(context, sizeof(OpData)); in Init()

123