Home
last modified time | relevance | path

Searched refs:kTfLiteArenaRwPersistent (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/lite/
Darena_planner.cc52 if (type == kTfLiteArenaRwPersistent) { in BasePointer()
327 if (tensor.allocation_type == kTfLiteArenaRwPersistent && in CalculateAllocations()
349 if (tensor.allocation_type == kTfLiteArenaRwPersistent) { in ResolveTensorAllocation()
Doptional_debug_tools.cc96 case kTfLiteArenaRwPersistent: in AllocTypeName()
Darena_planner_test.cc380 (*graph.tensors())[5].allocation_type = kTfLiteArenaRwPersistent; in TEST_F()
461 (*graph.tensors())[1].allocation_type = kTfLiteArenaRwPersistent; in TEST_F()
/external/tensorflow/tensorflow/lite/kernels/
Ddequantize.cc71 op_context.output->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
Ddensify.cc64 op_context.output->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
Dsvdf.cc206 float_weights_time->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
234 row_sums->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
Dnumeric_verify.cc123 op_context.output->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
Dbatch_matmul.cc192 scratch_buffer->allocation_type = kTfLiteArenaRwPersistent; in InitializeTemporaries()
278 row_sums->allocation_type = kTfLiteArenaRwPersistent; in InitializeTemporaries()
Dbasic_rnn.cc187 row_sums->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
Dbidirectional_sequence_rnn.cc302 fw_row_sums->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
318 bw_row_sums->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
Dunidirectional_sequence_rnn.cc194 row_sums->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
Dfully_connected.cc60 ledger->allocation_type = kTfLiteArenaRwPersistent; in CreateLedgerTensor()
330 row_sums->allocation_type = kTfLiteArenaRwPersistent; in PrepareImpl()
Dconv.cc490 hwcn_weights->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
584 row_sums->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
Dbidirectional_sequence_lstm.cc910 fw_row_sums->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
935 bw_row_sums->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
Dlstm.cc98 ledger->allocation_type = kTfLiteArenaRwPersistent; in make_ledger()
1612 row_sums->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
2262 tensor->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
Dunidirectional_sequence_lstm.cc1142 row_sums->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.cc719 if (tensor.allocation_type == kTfLiteArenaRwPersistent) { in ResetVariableTensors()
1286 allocation_type = kTfLiteArenaRwPersistent; in SetTensorParametersReadWrite()
1316 tensor->allocation_type == kTfLiteArenaRwPersistent || in ResizeTensorImpl()
1340 tensor->allocation_type == kTfLiteArenaRwPersistent) { in ResizeTensorImpl()
1598 tensor->allocation_type == kTfLiteArenaRwPersistent || in SetCustomAllocationForTensor()
/external/tensorflow/tensorflow/lite/c/
Dcommon.h333 kTfLiteArenaRwPersistent, enumerator