| /external/tensorflow/tensorflow/lite/kernels/ |
| D | kernel_util_test.cc | 152 input.allocation_type = kTfLiteArenaRw; in TEST_F()
169 filter.allocation_type = kTfLiteArenaRw; in TEST_F()
194 bias.allocation_type = kTfLiteArenaRw; in TEST_F()
214 output.allocation_type = kTfLiteArenaRw; in TEST_F()
258 input.allocation_type = kTfLiteArenaRw; in TEST_F()
275 filter.allocation_type = kTfLiteArenaRw; in TEST_F()
303 bias.allocation_type = kTfLiteArenaRw; in TEST_F()
323 output.allocation_type = kTfLiteArenaRw; in TEST_F()
|
| D | unidirectional_sequence_lstm.cc | 324 scratch_buffer->allocation_type = kTfLiteArenaRw; in Prepare()
349 input_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
360 activation_state_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
374 cell_state_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
393 scaling_factors->allocation_type = kTfLiteArenaRw; in Prepare()
406 prod_scaling_factors->allocation_type = kTfLiteArenaRw; in Prepare()
423 recovered_cell_weights->allocation_type = kTfLiteArenaRw; in Prepare()
|
| D | bidirectional_sequence_lstm.cc | 527 fw_scratch_buffer->allocation_type = kTfLiteArenaRw; in Prepare()
584 bw_scratch_buffer->allocation_type = kTfLiteArenaRw; in Prepare()
612 input_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
624 fw_activation_state_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
638 bw_activation_state_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
652 fw_cell_state_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
666 bw_cell_state_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
686 scaling_factors->allocation_type = kTfLiteArenaRw; in Prepare()
699 prod_scaling_factors->allocation_type = kTfLiteArenaRw; in Prepare()
716 recovered_cell_weights->allocation_type = kTfLiteArenaRw; in Prepare()
[all …]
|
| D | basic_rnn.cc | 111 input_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
121 hidden_state_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
133 scaling_factors->allocation_type = kTfLiteArenaRw; in Prepare()
|
| D | lstm.cc | 400 scratch_buffer->allocation_type = kTfLiteArenaRw; in Prepare()
423 input_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
433 activation_state_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
446 cell_state_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
463 scaling_factors->allocation_type = kTfLiteArenaRw; in Prepare()
475 prod_scaling_factors->allocation_type = kTfLiteArenaRw; in Prepare()
491 recovered_cell_weights->allocation_type = kTfLiteArenaRw; in Prepare()
|
| D | unidirectional_sequence_rnn.cc | 112 input_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
122 hidden_state_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
134 scaling_factors->allocation_type = kTfLiteArenaRw; in Prepare()
|
| D | bidirectional_sequence_rnn.cc | 191 input_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
203 fw_hidden_state_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
218 bw_hidden_state_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
234 scaling_factors->allocation_type = kTfLiteArenaRw; in Prepare()
249 aux_input_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
|
| D | svdf.cc | 198 scratch_tensor->allocation_type = kTfLiteArenaRw; in Prepare()
208 input_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
219 scaling_factors->allocation_type = kTfLiteArenaRw; in Prepare()
|
| D | reduce.cc | 169 scratch_tensor->allocation_type = kTfLiteArenaRw; in InitializeTemporaries()
221 resolved_axis->allocation_type = kTfLiteArenaRw; in PrepareSimple()
254 temp_sum->allocation_type = kTfLiteArenaRw; in PrepareMeanOrSum()
|
| D | detection_postprocess.cc | 196 decoded_boxes->allocation_type = kTfLiteArenaRw; in Prepare()
203 scores->allocation_type = kTfLiteArenaRw; in Prepare()
212 active_candidate->allocation_type = kTfLiteArenaRw; in Prepare()
|
| D | conv.cc | 353 im2col->allocation_type = kTfLiteArenaRw; in Prepare()
390 input_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
402 scaling_factors->allocation_type = kTfLiteArenaRw; in Prepare()
|
| D | fully_connected.cc | 143 input_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
154 scaling_factors->allocation_type = kTfLiteArenaRw; in Prepare()
|
| /external/tensorflow/tensorflow/lite/ |
| D | arena_planner.cc | 47 if (type == kTfLiteArenaRw) { in BasePointer()
249 if (tensor.allocation_type == kTfLiteArenaRw) { in ResolveTensorAllocation()
266 if (tensor.allocation_type == kTfLiteArenaRw) { in CalculateTensorAllocation()
279 if (tensor.allocation_type == kTfLiteArenaRw) { in CalculateTensorDeallocation()
|
| D | model_test.cc | 150 ASSERT_EQ(i1->allocation_type, kTfLiteArenaRw); in TEST()
154 ASSERT_EQ(o0->allocation_type, kTfLiteArenaRw); in TEST()
158 ASSERT_EQ(o1->allocation_type, kTfLiteArenaRw); in TEST()
|
| D | optional_debug_tools.cc | 71 case kTfLiteArenaRw: in AllocTypeName()
|
| D | arena_planner_test.cc | 88 tensors_.back().allocation_type = kTfLiteArenaRw; in TestGraph()
|
| D | interpreter_test.cc | 559 tmp->allocation_type = kTfLiteArenaRw; in TEST()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | c_api_internal_test.cc | 94 t.allocation_type = kTfLiteArenaRw; in TEST()
|
| D | c_api_internal.h | 274 kTfLiteArenaRw, enumerator
|
| /external/tensorflow/tensorflow/lite/experimental/micro/ |
| D | simple_tensor_allocator.cc | 124 result->allocation_type = kTfLiteArenaRw; in AllocateTensor()
|
| /external/tensorflow/tensorflow/lite/experimental/writer/ |
| D | writer_lib.cc | 154 if (tensor->allocation_type != kTfLiteArenaRw && in ExportTensors()
|
| /external/tensorflow/tensorflow/lite/core/ |
| D | subgraph.cc | 890 TfLiteAllocationType allocation_type = kTfLiteArenaRw; in SetTensorParametersReadWrite()
925 if (tensor->allocation_type == kTfLiteArenaRw || in ResizeTensorImpl()
|
| /external/tensorflow/tensorflow/lite/g3doc/guide/ |
| D | ops_custom.md | 157 temp_tensor->allocation_type = kTfLiteArenaRw;
|