Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/lite/
Dcontext.c59 if (t->allocation_type == kTfLiteDynamic && t->data.raw) { in TfLiteTensorFree()
69 size_t size, TfLiteAllocationType allocation_type, in TfLiteTensorReset() argument
78 tensor->allocation_type = allocation_type; in TfLiteTensorReset()
83 if (tensor->allocation_type != kTfLiteDynamic) { in TfLiteTensorRealloc()
Dstring_util_test.cc30 t0->allocation_type = kTfLiteDynamic; in TEST()
34 t1->allocation_type = kTfLiteDynamic; in TEST()
81 t0->allocation_type = kTfLiteDynamic; in TEST()
104 t0->allocation_type = kTfLiteDynamic; in TEST()
Darena_planner.cc187 if (tensor.allocation_type == kTfLiteArenaRw) { in ResolveTensorAllocation()
195 if (tensor.allocation_type == kTfLiteArenaRwPersistent) { in ResolveTensorAllocation()
204 if (tensor.allocation_type == kTfLiteArenaRw) { in CalculateTensorAllocation()
209 if (tensor.allocation_type == kTfLiteArenaRwPersistent) { in CalculateTensorAllocation()
219 if (tensor.allocation_type == kTfLiteArenaRw) { in CalculateTensorDeallocation()
Dmodel_test.cc144 ASSERT_EQ(i0->allocation_type, kTfLiteMmapRo); in TEST()
148 ASSERT_EQ(i1->allocation_type, kTfLiteArenaRw); in TEST()
152 ASSERT_EQ(o0->allocation_type, kTfLiteArenaRw); in TEST()
156 ASSERT_EQ(o1->allocation_type, kTfLiteArenaRw); in TEST()
Darena_planner_test.cc86 tensors_.back().allocation_type = kTfLiteArenaRw; in TestGraph()
163 planner_->BasePointer(tensor.allocation_type); in GetOffset()
318 (*graph.tensors())[1].allocation_type = kTfLiteArenaRwPersistent; in TEST_F()
347 (*graph.tensors())[1].allocation_type = kTfLiteDynamic; in TEST_F()
Dcontext.h185 TfLiteAllocationType allocation_type; member
205 size_t size, TfLiteAllocationType allocation_type,
Dinterpreter.cc324 if (tensor.allocation_type == kTfLiteDynamic) { in HasDynamicTensor()
546 if (tensor->allocation_type == kTfLiteArenaRw || in ResizeTensorImpl()
547 tensor->allocation_type == kTfLiteDynamic) { in ResizeTensorImpl()
564 if (tensor->allocation_type != kTfLiteDynamic) { in ResizeTensorImpl()
Doptional_debug_tools.cc84 TensorTypeName(tensor->type), AllocTypeName(tensor->allocation_type), in PrintInterpreterState()
Dnnapi_delegate.cc119 if (tensor->allocation_type == kTfLiteMmapRo) { in addTensorOperands()
Dinterpreter_test.cc309 tensor->allocation_type = kTfLiteDynamic; in TEST()
376 tmp->allocation_type = kTfLiteArenaRw; in TEST()
/external/tensorflow/tensorflow/contrib/lite/kernels/
Dkernel_util.h57 return tensor->allocation_type == kTfLiteMmapRo; in IsConstantTensor()
63 return tensor->allocation_type == kTfLiteDynamic; in IsDynamicTensor()
68 if (tensor->allocation_type != kTfLiteDynamic) { in SetTensorToDynamic()
69 tensor->allocation_type = kTfLiteDynamic; in SetTensorToDynamic()
Dkernel_util_test.cc33 tensor1_.allocation_type = kTfLiteMmapRo; in KernelUtilTest()
34 tensor2_.allocation_type = kTfLiteMmapRo; in KernelUtilTest()
Dmean.cc142 scratch_tensor->allocation_type = kTfLiteArenaRw; in InitializeTemporaries()
169 resolved_axis->allocation_type = kTfLiteArenaRw; in Prepare()
Dsvdf.cc97 state->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
117 scratch_tensor->allocation_type = kTfLiteArenaRw; in Prepare()
Dbidirectional_sequence_rnn.cc103 fw_hidden_state->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
104 bw_hidden_state->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
Dbasic_rnn.cc74 hidden_state->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
Dunidirectional_sequence_rnn.cc79 hidden_state->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
Dembedding_lookup_sparse.cc111 output->allocation_type = kTfLiteDynamic; in Prepare()
Dconv.cc246 im2col->allocation_type = kTfLiteArenaRw; in Prepare()
266 hwcn_weights->allocation_type = kTfLiteDynamic; in Prepare()
Dlstm.cc281 output_state->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
282 cell_state->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
Dunidirectional_sequence_lstm.cc283 output_state->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
284 cell_state->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
/external/mesa3d/include/CL/
Dcl_ext.h282 cl_uint allocation_type; member