Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/lite/kernels/
Dkernel_util_test.cc35 tensor1_.allocation_type = kTfLiteMmapRo; in KernelUtilTest()
36 tensor2_.allocation_type = kTfLiteMmapRo; in KernelUtilTest()
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()
Dkernel_util.h70 return tensor->allocation_type == kTfLiteMmapRo; in IsConstantTensor()
76 return tensor->allocation_type == kTfLiteDynamic; in IsDynamicTensor()
81 if (tensor->allocation_type != kTfLiteDynamic) { in SetTensorToDynamic()
82 tensor->allocation_type = kTfLiteDynamic; in SetTensorToDynamic()
Dunidirectional_sequence_lstm.cc324 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()
Dbidirectional_sequence_lstm.cc527 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 …]
Dlstm.cc400 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()
728 tensor->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
Dbasic_rnn.cc111 input_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
121 hidden_state_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
133 scaling_factors->allocation_type = kTfLiteArenaRw; in Prepare()
Dsvdf.cc198 scratch_tensor->allocation_type = kTfLiteArenaRw; in Prepare()
208 input_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
219 scaling_factors->allocation_type = kTfLiteArenaRw; in Prepare()
234 float_weights_time->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
Dunidirectional_sequence_rnn.cc112 input_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
122 hidden_state_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
134 scaling_factors->allocation_type = kTfLiteArenaRw; in Prepare()
Dbidirectional_sequence_rnn.cc191 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()
Dconv.cc353 im2col->allocation_type = kTfLiteArenaRw; in Prepare()
373 hwcn_weights->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
390 input_quantized->allocation_type = kTfLiteArenaRw; in Prepare()
402 scaling_factors->allocation_type = kTfLiteArenaRw; in Prepare()
Ddequantize.cc68 op_context.output->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
/external/tensorflow/tensorflow/lite/
Dstring_util_test.cc30 t0->allocation_type = kTfLiteDynamic; in TEST()
34 t1->allocation_type = kTfLiteDynamic; in TEST()
96 t0->allocation_type = kTfLiteDynamic; in TEST()
119 t0->allocation_type = kTfLiteDynamic; in TEST()
132 t0->allocation_type = kTfLiteDynamic; in TEST()
Darena_planner.cc249 if (tensor.allocation_type == kTfLiteArenaRw) { in ResolveTensorAllocation()
257 if (tensor.allocation_type == kTfLiteArenaRwPersistent) { in ResolveTensorAllocation()
266 if (tensor.allocation_type == kTfLiteArenaRw) { in CalculateTensorAllocation()
270 if (tensor.allocation_type == kTfLiteArenaRwPersistent) { in CalculateTensorAllocation()
279 if (tensor.allocation_type == kTfLiteArenaRw) { in CalculateTensorDeallocation()
Dmodel_test.cc146 ASSERT_EQ(i0->allocation_type, kTfLiteMmapRo); in TEST()
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()
Darena_planner_test.cc88 tensors_.back().allocation_type = kTfLiteArenaRw; in TestGraph()
188 planner_->BasePointer(tensor.allocation_type); in GetOffset()
364 (*graph.tensors())[1].allocation_type = kTfLiteArenaRwPersistent; in TEST_F()
395 (*graph.tensors())[1].allocation_type = kTfLiteDynamic; in TEST_F()
Doptional_debug_tools.cc93 AllocTypeName(tensor->allocation_type), tensor->bytes, in PrintInterpreterState()
/external/tensorflow/tensorflow/lite/c/
Dc_api_internal.c88 if (t->allocation_type == kTfLiteDynamic && t->data.raw) { in TfLiteTensorDataFree()
122 size_t size, TfLiteAllocationType allocation_type, in TfLiteTensorReset() argument
132 tensor->allocation_type = allocation_type; in TfLiteTensorReset()
141 if (tensor->allocation_type != kTfLiteDynamic) { in TfLiteTensorRealloc()
Dc_api_internal.h304 TfLiteAllocationType allocation_type; member
352 size_t size, TfLiteAllocationType allocation_type,
Dc_api_internal_test.cc94 t.allocation_type = kTfLiteArenaRw; in TEST()
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.cc63 if (tensor.allocation_type == kTfLiteDynamic) { in HasDynamicTensorImpl()
509 TF_LITE_ENSURE_EQ(context_, tensor.allocation_type, in ResetVariableTensors()
851 tensor.allocation_type = kTfLiteMmapRo; in SetTensorParametersReadOnly()
890 TfLiteAllocationType allocation_type = kTfLiteArenaRw; in SetTensorParametersReadWrite() local
897 allocation_type = kTfLiteDynamic; in SetTensorParametersReadWrite()
899 allocation_type = kTfLiteArenaRwPersistent; in SetTensorParametersReadWrite()
905 /*buffer=*/nullptr, required_bytes, allocation_type, in SetTensorParametersReadWrite()
925 if (tensor->allocation_type == kTfLiteArenaRw || in ResizeTensorImpl()
926 tensor->allocation_type == kTfLiteDynamic || in ResizeTensorImpl()
927 tensor->allocation_type == kTfLiteArenaRwPersistent) { in ResizeTensorImpl()
[all …]
/external/tensorflow/tensorflow/lite/experimental/writer/
Dwriter_lib.cc154 if (tensor->allocation_type != kTfLiteArenaRw && in ExportTensors()
155 tensor->allocation_type != kTfLiteMmapRo && in ExportTensors()
156 tensor->allocation_type != kTfLiteArenaRwPersistent) in ExportTensors()
160 if (tensor->allocation_type == kTfLiteMmapRo) { in ExportTensors()
/external/syzkaller/sys/windows/
Dsys.txt11 VirtualAlloc(lpAddress vma, dwSize len[lpAddress], flAllocationType flags[allocation_type], flProte…
29 allocation_type = MEM_COMMIT, MEM_RESERVE, MEM_RESET, MEM_RESET_UNDO, MEM_LARGE_PAGES, MEM_PHYSICAL…
/external/tensorflow/tensorflow/lite/experimental/micro/
Dsimple_tensor_allocator.cc101 result->allocation_type = kTfLiteMmapRo; in AllocateTensor()
124 result->allocation_type = kTfLiteArenaRw; in AllocateTensor()
/external/tensorflow/tensorflow/lite/delegates/flex/
Dbuffer_map_test.cc42 tensor->allocation_type = kTfLiteDynamic; in MakeLiteTensor()
61 tensor->allocation_type = kTfLiteDynamic; in MakeLiteTensor()
/external/mesa3d/include/CL/
Dcl_ext.h282 cl_uint allocation_type; member

12