Home
last modified time | relevance | path

Searched refs:allocation_type (Results 1 – 25 of 82) sorted by relevance

1234

/external/tensorflow/tensorflow/lite/c/
Dcommon.c84 if (t->allocation_type == kTfLiteDynamic || in TfLiteTensorDataFree()
85 t->allocation_type == kTfLitePersistentRo) { in TfLiteTensorDataFree()
159 size_t size, TfLiteAllocationType allocation_type, in TfLiteTensorReset() argument
169 tensor->allocation_type = allocation_type; in TfLiteTensorReset()
178 if (tensor->allocation_type != kTfLiteDynamic && in TfLiteTensorRealloc()
179 tensor->allocation_type != kTfLitePersistentRo) { in TfLiteTensorRealloc()
Dcommon_test.cc102 t.allocation_type = kTfLiteArenaRw; in TEST()
118 t.allocation_type = kTfLiteArenaRw; in TEST()
Dcommon.h399 TfLiteAllocationType allocation_type; member
530 TfLiteAllocationType allocation_type; member
598 size_t size, TfLiteAllocationType allocation_type,
/external/tensorflow/tensorflow/lite/kernels/
Dkernel_util_test.cc62 tensor1_.allocation_type = kTfLiteMmapRo; in KernelUtilTest()
63 tensor2_.allocation_type = kTfLiteMmapRo; in KernelUtilTest()
64 tensor3_.allocation_type = kTfLiteMmapRo; in KernelUtilTest()
274 input.allocation_type = kTfLiteArenaRw; in TEST_F()
291 filter.allocation_type = kTfLiteArenaRw; in TEST_F()
316 bias.allocation_type = kTfLiteArenaRw; in TEST_F()
336 output.allocation_type = kTfLiteArenaRw; in TEST_F()
380 input.allocation_type = kTfLiteArenaRw; in TEST_F()
397 filter.allocation_type = kTfLiteArenaRw; in TEST_F()
419 bias.allocation_type = kTfLiteArenaRw; in TEST_F()
[all …]
Dkernel_util.h179 return tensor->allocation_type == kTfLiteMmapRo; in IsConstantTensor()
185 return tensor->allocation_type == kTfLiteDynamic; in IsDynamicTensor()
190 if (tensor->allocation_type != kTfLiteDynamic) { in SetTensorToDynamic()
191 tensor->allocation_type = kTfLiteDynamic; in SetTensorToDynamic()
198 if (tensor->allocation_type != kTfLitePersistentRo) { in SetTensorToPersistentRo()
199 tensor->allocation_type = kTfLitePersistentRo; in SetTensorToPersistentRo()
Dsvdf.cc163 scratch_tensor->allocation_type = kTfLiteArenaRw; in Prepare()
176 input_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
189 scaling_factors->allocation_type = kTfLiteArenaRw; in Prepare()
206 float_weights_time->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
220 zero_points->allocation_type = kTfLiteArenaRw; in Prepare()
234 row_sums->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
253 output_temp->allocation_type = kTfLiteArenaRw; in Prepare()
Dbidirectional_sequence_lstm.cc575 fw_scratch_buffer->allocation_type = kTfLiteArenaRw; in Prepare()
637 bw_scratch_buffer->allocation_type = kTfLiteArenaRw; in Prepare()
669 input_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
683 fw_activation_state_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
699 bw_activation_state_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
715 fw_cell_state_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
731 bw_cell_state_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
753 input_sf->allocation_type = kTfLiteArenaRw; in Prepare()
768 aux_input_sf->allocation_type = kTfLiteArenaRw; in Prepare()
782 output_state_sf->allocation_type = kTfLiteArenaRw; in Prepare()
[all …]
Dbasic_rnn.cc121 input_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
132 hidden_state_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
146 scaling_factors->allocation_type = kTfLiteArenaRw; in Prepare()
159 accum_scratch->allocation_type = kTfLiteArenaRw; in Prepare()
174 zero_points->allocation_type = kTfLiteArenaRw; in Prepare()
187 row_sums->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
Dunidirectional_sequence_rnn.cc128 input_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
139 hidden_state_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
153 scaling_factors->allocation_type = kTfLiteArenaRw; in Prepare()
166 accum_scratch->allocation_type = kTfLiteArenaRw; in Prepare()
181 zero_points->allocation_type = kTfLiteArenaRw; in Prepare()
194 row_sums->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
Dbatch_matmul.cc167 scratch_buffer->allocation_type = kTfLiteArenaRw; in InitializeTemporaries()
192 scratch_buffer->allocation_type = kTfLiteArenaRwPersistent; in InitializeTemporaries()
194 scratch_buffer->allocation_type = kTfLiteArenaRw; in InitializeTemporaries()
223 input_quantized->allocation_type = kTfLiteArenaRw; in InitializeTemporaries()
235 scaling_factors->allocation_type = kTfLiteArenaRw; in InitializeTemporaries()
250 accum_scratch->allocation_type = kTfLiteArenaRw; in InitializeTemporaries()
266 input_offsets->allocation_type = kTfLiteArenaRw; in InitializeTemporaries()
278 row_sums->allocation_type = kTfLiteArenaRwPersistent; in InitializeTemporaries()
Dbidirectional_sequence_rnn.cc206 input_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
220 fw_hidden_state_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
237 bw_hidden_state_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
254 scaling_factors->allocation_type = kTfLiteArenaRw; in Prepare()
268 accum_scratch->allocation_type = kTfLiteArenaRw; in Prepare()
286 zero_points->allocation_type = kTfLiteArenaRw; in Prepare()
302 fw_row_sums->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
318 bw_row_sums->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
335 aux_input_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
Dunidirectional_sequence_lstm.cc957 scratch_buffer->allocation_type = kTfLiteArenaRw; in Prepare()
984 input_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
997 output_state_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
1013 cell_state_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
1034 input_sf->allocation_type = kTfLiteArenaRw; in Prepare()
1049 output_state_sf->allocation_type = kTfLiteArenaRw; in Prepare()
1063 prod_scaling_factors->allocation_type = kTfLiteArenaRw; in Prepare()
1082 recovered_cell_weights->allocation_type = kTfLiteArenaRw; in Prepare()
1100 accum_scratch->allocation_type = kTfLiteArenaRw; in Prepare()
1116 input_zp->allocation_type = kTfLiteArenaRw; in Prepare()
[all …]
/external/tensorflow/tensorflow/lite/experimental/resource/
Dresource_variable_test.cc54 EXPECT_EQ(kTfLiteDynamic, value->allocation_type); in TEST()
80 EXPECT_EQ(kTfLiteDynamic, value->allocation_type); in TEST()
107 EXPECT_EQ(kTfLiteDynamic, value->allocation_type); in TEST()
119 EXPECT_EQ(kTfLiteDynamic, value->allocation_type); in TEST()
148 EXPECT_EQ(kTfLiteDynamic, value->allocation_type); in TEST()
160 EXPECT_EQ(kTfLiteDynamic, value->allocation_type); in TEST()
/external/tensorflow/tensorflow/lite/
Darena_planner.cc73 if (tensor.allocation_type == kTfLiteArenaRw) { in ResetAllocationsAfter()
232 if (tensor.allocation_type == kTfLiteArenaRw) { in ReleaseNonPersistentMemory()
245 if (tensor.allocation_type == kTfLiteArenaRw) { in AcquireNonPersistentMemory()
311 if (tensor.allocation_type == kTfLiteArenaRw && in CalculateAllocations()
320 if (tensor.allocation_type == kTfLiteArenaRw) { in CalculateAllocations()
327 if (tensor.allocation_type == kTfLiteArenaRwPersistent && in CalculateAllocations()
341 if (tensor.allocation_type == kTfLiteArenaRw) { in ResolveTensorAllocation()
349 if (tensor.allocation_type == kTfLiteArenaRwPersistent) { in ResolveTensorAllocation()
Dstring_util_test.cc34 t0->allocation_type = kTfLiteDynamic; in TEST()
38 t1->allocation_type = kTfLiteDynamic; in TEST()
109 t0->allocation_type = kTfLiteDynamic; in TEST()
133 t0->allocation_type = kTfLiteDynamic; in TEST()
158 t0->allocation_type = kTfLiteDynamic; in TEST()
171 t0->allocation_type = kTfLiteDynamic; in TEST()
Darena_planner_test.cc99 tensors_.back().allocation_type = kTfLiteArenaRw; in TestGraph()
218 planner_->BasePointer(tensor.allocation_type); in GetOffset()
380 (*graph.tensors())[5].allocation_type = kTfLiteArenaRwPersistent; in TEST_F()
461 (*graph.tensors())[1].allocation_type = kTfLiteArenaRwPersistent; in TEST_F()
492 (*graph.tensors())[1].allocation_type = kTfLiteDynamic; in TEST_F()
Dmodel_test.cc218 ASSERT_EQ(i0->allocation_type, kTfLiteMmapRo); in TEST()
222 ASSERT_EQ(i1->allocation_type, kTfLiteArenaRw); in TEST()
226 ASSERT_EQ(o0->allocation_type, kTfLiteArenaRw); in TEST()
230 ASSERT_EQ(o1->allocation_type, kTfLiteArenaRw); in TEST()
422 ASSERT_EQ(t1->allocation_type, kTfLiteMmapRo); in TEST()
/external/tensorflow/tensorflow/lite/micro/
Drecording_micro_allocator.cc49 RecordedAllocationType allocation_type) const { in GetRecordedAllocation()
50 switch (allocation_type) { in GetRecordedAllocation()
67 allocation_type); in GetRecordedAllocation()
117 RecordedAllocationType allocation_type, const char* allocation_name, in PrintRecordedAllocation() argument
120 RecordedAllocation allocation = GetRecordedAllocation(allocation_type); in PrintRecordedAllocation()
Drecording_micro_allocator.h63 RecordedAllocationType allocation_type) const;
102 void PrintRecordedAllocation(RecordedAllocationType allocation_type,
Dmemory_arena_threshold_test.cc104 void EnsureAllocatedSizeThreshold(const char* allocation_type, size_t actual, in EnsureAllocatedSizeThreshold() argument
113 "%s threshold failed: %d != %d", allocation_type, in EnsureAllocatedSizeThreshold()
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.cc105 if (tensor.allocation_type == kTfLiteDynamic) { in HasDynamicTensorImpl()
719 if (tensor.allocation_type == kTfLiteArenaRwPersistent) { in ResetVariableTensors()
728 TF_LITE_ENSURE_EQ(&context_, tensor.allocation_type, kTfLiteCustom); in ResetVariableTensors()
986 tensor_at_index->allocation_type == kTfLiteCustom); in PrepareOpsAndTensors()
1232 tensor.allocation_type = kTfLiteMmapRo; in SetTensorParametersReadOnly()
1276 TfLiteAllocationType allocation_type = kTfLiteArenaRw; in SetTensorParametersReadWrite() local
1284 allocation_type = kTfLiteDynamic; in SetTensorParametersReadWrite()
1286 allocation_type = kTfLiteArenaRwPersistent; in SetTensorParametersReadWrite()
1292 /*buffer=*/nullptr, required_bytes, allocation_type, in SetTensorParametersReadWrite()
1314 if (tensor->allocation_type == kTfLiteArenaRw || in ResizeTensorImpl()
[all …]
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dutil_test.cc39 tensor.allocation_type = kTfLiteArenaRw; in SetUp()
94 tensors_[1].allocation_type = kTfLiteMmapRo; in TEST_F()
/external/tensorflow/tensorflow/lite/delegates/hexagon/
Dhexagon_delegate_kernel.cc104 if (tensor->allocation_type != kTfLiteMmapRo) { in Eval()
129 if (tensor->allocation_type != kTfLiteMmapRo) { in Eval()
223 if (tensor->allocation_type != kTfLiteMmapRo && tensor->dims->size > 4) { in Prepare()
266 if (input_tensor.allocation_type == kTfLiteMmapRo) { in BuildGraph()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dquantization_util_test.cc94 result.allocation_type = kTfLiteMemNone; in CreateQuantizedTensor()
113 result.allocation_type = kTfLiteMemNone; in CreateQuantizedTensor()
131 result.allocation_type = kTfLiteMemNone; in CreateTensor()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dpad_test.cc109 tensors[1].allocation_type = kTfLiteMmapRo; in TestPadFloat()
137 tensors[1].allocation_type = kTfLiteMmapRo; in TestPadV2Float()
167 tensors[1].allocation_type = kTfLiteMmapRo; in TestPadQuantized()
208 tensors[1].allocation_type = kTfLiteMmapRo; in TestPadV2Quantized()

1234