/external/tensorflow/tensorflow/lite/kernels/ |
D | kernel_util_test.cc | 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() 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 …]
|
D | bidirectional_sequence_lstm.cc | 575 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 …]
|
D | basic_rnn.cc | 121 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()
|
D | svdf.cc | 163 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()
|
D | unidirectional_sequence_rnn.cc | 128 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()
|
D | unidirectional_sequence_lstm.cc | 957 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 …]
|
D | batch_matmul.cc | 167 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()
|
D | bidirectional_sequence_rnn.cc | 206 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()
|
D | lstm.cc | 1413 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 …]
|
D | conv.cc | 396 (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()
|
D | rfft2d.cc | 83 fft_integer_working_area->allocation_type = kTfLiteArenaRw; in InitTemporaryTensors() 100 fft_double_working_area->allocation_type = kTfLiteArenaRw; in InitTemporaryTensors()
|
D | depthwise_conv.cc | 237 input_quantized->allocation_type = kTfLiteArenaRw; in Prepare() 250 scaling_factors->allocation_type = kTfLiteArenaRw; in Prepare() 265 input_offsets->allocation_type = kTfLiteArenaRw; in Prepare()
|
D | fully_connected.cc | 276 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()
|
D | detection_postprocess.cc | 213 decoded_boxes->allocation_type = kTfLiteArenaRw; in Prepare() 220 scores->allocation_type = kTfLiteArenaRw; in Prepare() 229 active_candidate->allocation_type = kTfLiteArenaRw; in Prepare()
|
D | reduce.cc | 177 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/ |
D | arena_planner.cc | 55 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()
|
D | optional_debug_tools.cc | 94 case kTfLiteArenaRw: in AllocTypeName()
|
D | model_test.cc | 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()
|
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/mlir2exec/ |
D | tfl_while_op.mlir | 15 // 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/ |
D | common_test.cc | 102 t.allocation_type = kTfLiteArenaRw; in TEST() 118 t.allocation_type = kTfLiteArenaRw; in TEST()
|
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/ |
D | util_test.cc | 39 tensor.allocation_type = kTfLiteArenaRw; in SetUp()
|
/external/tensorflow/tensorflow/lite/experimental/kernels/ |
D | unidirectional_sequence_gru.cc | 203 activation->allocation_type = kTfLiteArenaRw; in Prepare() 215 concat->allocation_type = kTfLiteArenaRw; in Prepare()
|
/external/tensorflow/tensorflow/lite/experimental/resource/ |
D | resource_variable_test.cc | 47 InitTensor(shape, kTfLiteArenaRw, 1.0f, &tensor); in TEST()
|
/external/tensorflow/tensorflow/lite/core/ |
D | subgraph.cc | 1276 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/ |
D | utils_test.cc | 68 EXPECT_EQ(new_tensor->allocation_type, kTfLiteArenaRw); in TEST()
|