Home
last modified time | relevance | path

Searched refs:kTfLiteArenaRw (Results 1 – 25 of 33) sorted by relevance

12

/external/tensorflow/tensorflow/lite/kernels/
Dkernel_util_test.cc274 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()
439 output.allocation_type = kTfLiteArenaRw; in TEST_F()
488 input.allocation_type = kTfLiteArenaRw; in TEST_F()
505 filter.allocation_type = kTfLiteArenaRw; in TEST_F()
[all …]
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()
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()
220 zero_points->allocation_type = kTfLiteArenaRw; in Prepare()
253 output_temp->allocation_type = kTfLiteArenaRw; 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()
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 …]
Dbatch_matmul.cc167 scratch_buffer->allocation_type = kTfLiteArenaRw; 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()
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()
335 aux_input_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
Dlstm.cc1413 scratch_buffer->allocation_type = kTfLiteArenaRw; in Prepare()
1443 input_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
1456 output_state_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
1472 cell_state_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
1492 input_sf->allocation_type = kTfLiteArenaRw; in Prepare()
1507 output_state_sf->allocation_type = kTfLiteArenaRw; in Prepare()
1521 prod_scaling_factors->allocation_type = kTfLiteArenaRw; in Prepare()
1540 recovered_cell_weights->allocation_type = kTfLiteArenaRw; in Prepare()
1558 accum_scratch->allocation_type = kTfLiteArenaRw; in Prepare()
1574 input_zp->allocation_type = kTfLiteArenaRw; in Prepare()
[all …]
Dconv.cc396 (filter->allocation_type != kTfLiteArenaRw) && in Prepare()
470 im2col->allocation_type = kTfLiteArenaRw; in Prepare()
509 input_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
523 scaling_factors->allocation_type = kTfLiteArenaRw; in Prepare()
542 accum_scratch->allocation_type = kTfLiteArenaRw; in Prepare()
567 input_offsets->allocation_type = kTfLiteArenaRw; in Prepare()
Drfft2d.cc83 fft_integer_working_area->allocation_type = kTfLiteArenaRw; in InitTemporaryTensors()
100 fft_double_working_area->allocation_type = kTfLiteArenaRw; in InitTemporaryTensors()
Ddepthwise_conv.cc237 input_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
250 scaling_factors->allocation_type = kTfLiteArenaRw; in Prepare()
265 input_offsets->allocation_type = kTfLiteArenaRw; in Prepare()
Dfully_connected.cc276 input_quantized->allocation_type = kTfLiteArenaRw; in PrepareImpl()
287 scaling_factors->allocation_type = kTfLiteArenaRw; in PrepareImpl()
302 accum_scratch->allocation_type = kTfLiteArenaRw; in PrepareImpl()
318 input_offsets->allocation_type = kTfLiteArenaRw; in PrepareImpl()
Ddetection_postprocess.cc213 decoded_boxes->allocation_type = kTfLiteArenaRw; in Prepare()
220 scores->allocation_type = kTfLiteArenaRw; in Prepare()
229 active_candidate->allocation_type = kTfLiteArenaRw; in Prepare()
Dreduce.cc177 scratch_tensor->allocation_type = kTfLiteArenaRw; in InitializeTemporaries()
240 resolved_axis->allocation_type = kTfLiteArenaRw; in PrepareSimple()
284 temp_sum->allocation_type = kTfLiteArenaRw; in PrepareMeanOrSum()
/external/tensorflow/tensorflow/lite/
Darena_planner.cc55 if (type == kTfLiteArenaRw) { in BasePointer()
73 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()
341 if (tensor.allocation_type == kTfLiteArenaRw) { in ResolveTensorAllocation()
Doptional_debug_tools.cc94 case kTfLiteArenaRw: in AllocTypeName()
Dmodel_test.cc222 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()
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/mlir2exec/
Dtfl_while_op.mlir15 // CHECK-NEXT: Tensor 3 tfl.while kTfLiteInt32 kTfLiteArenaRw 4 bytes
16 // CHECK-NEXT: Tensor 4 result kTfLiteFloat32 kTfLiteArenaRw 4 bytes
/external/tensorflow/tensorflow/lite/c/
Dcommon_test.cc102 t.allocation_type = kTfLiteArenaRw; in TEST()
118 t.allocation_type = kTfLiteArenaRw; in TEST()
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dutil_test.cc39 tensor.allocation_type = kTfLiteArenaRw; in SetUp()
/external/tensorflow/tensorflow/lite/experimental/kernels/
Dunidirectional_sequence_gru.cc203 activation->allocation_type = kTfLiteArenaRw; in Prepare()
215 concat->allocation_type = kTfLiteArenaRw; in Prepare()
/external/tensorflow/tensorflow/lite/experimental/resource/
Dresource_variable_test.cc47 InitTensor(shape, kTfLiteArenaRw, 1.0f, &tensor); in TEST()
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.cc1276 TfLiteAllocationType allocation_type = kTfLiteArenaRw; in SetTensorParametersReadWrite()
1314 if (tensor->allocation_type == kTfLiteArenaRw || in ResizeTensorImpl()
1339 if (tensor->allocation_type == kTfLiteArenaRw || in ResizeTensorImpl()
1597 (tensor->allocation_type == kTfLiteArenaRw || in SetCustomAllocationForTensor()
/external/tensorflow/tensorflow/lite/delegates/
Dutils_test.cc68 EXPECT_EQ(new_tensor->allocation_type, kTfLiteArenaRw); in TEST()

12