Searched refs:allocation_type (Results 1 – 22 of 22) sorted by relevance
/external/tensorflow/tensorflow/contrib/lite/ |
D | context.c | 59 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()
|
D | string_util_test.cc | 30 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()
|
D | arena_planner.cc | 187 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()
|
D | model_test.cc | 144 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()
|
D | arena_planner_test.cc | 86 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()
|
D | context.h | 185 TfLiteAllocationType allocation_type; member 205 size_t size, TfLiteAllocationType allocation_type,
|
D | interpreter.cc | 324 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()
|
D | optional_debug_tools.cc | 84 TensorTypeName(tensor->type), AllocTypeName(tensor->allocation_type), in PrintInterpreterState()
|
D | nnapi_delegate.cc | 119 if (tensor->allocation_type == kTfLiteMmapRo) { in addTensorOperands()
|
D | interpreter_test.cc | 309 tensor->allocation_type = kTfLiteDynamic; in TEST() 376 tmp->allocation_type = kTfLiteArenaRw; in TEST()
|
/external/tensorflow/tensorflow/contrib/lite/kernels/ |
D | kernel_util.h | 57 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()
|
D | kernel_util_test.cc | 33 tensor1_.allocation_type = kTfLiteMmapRo; in KernelUtilTest() 34 tensor2_.allocation_type = kTfLiteMmapRo; in KernelUtilTest()
|
D | mean.cc | 142 scratch_tensor->allocation_type = kTfLiteArenaRw; in InitializeTemporaries() 169 resolved_axis->allocation_type = kTfLiteArenaRw; in Prepare()
|
D | svdf.cc | 97 state->allocation_type = kTfLiteArenaRwPersistent; in Prepare() 117 scratch_tensor->allocation_type = kTfLiteArenaRw; in Prepare()
|
D | bidirectional_sequence_rnn.cc | 103 fw_hidden_state->allocation_type = kTfLiteArenaRwPersistent; in Prepare() 104 bw_hidden_state->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
|
D | basic_rnn.cc | 74 hidden_state->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
|
D | unidirectional_sequence_rnn.cc | 79 hidden_state->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
|
D | embedding_lookup_sparse.cc | 111 output->allocation_type = kTfLiteDynamic; in Prepare()
|
D | conv.cc | 246 im2col->allocation_type = kTfLiteArenaRw; in Prepare() 266 hwcn_weights->allocation_type = kTfLiteDynamic; in Prepare()
|
D | lstm.cc | 281 output_state->allocation_type = kTfLiteArenaRwPersistent; in Prepare() 282 cell_state->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
|
D | unidirectional_sequence_lstm.cc | 283 output_state->allocation_type = kTfLiteArenaRwPersistent; in Prepare() 284 cell_state->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
|
/external/mesa3d/include/CL/ |
D | cl_ext.h | 282 cl_uint allocation_type; member
|