Home
last modified time | relevance | path

Searched refs:allocate_entry_params (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dcpu_function_runtime.cc59 bool allocate_entry_params) { in AlignedBufferBytes() argument
64 (buffer_infos[i].is_entry_parameter() && allocate_entry_params); in AlignedBufferBytes()
74 bool allocate_entry_params, void** bufs, in MallocContiguousBuffers() argument
77 AlignedBufferBytes(buffer_infos, n, allocate_entry_params); in MallocContiguousBuffers()
91 (buffer_infos[i].is_entry_parameter() && allocate_entry_params); in MallocContiguousBuffers()
Dcpu_function_runtime.h160 bool allocate_entry_params);
175 bool allocate_entry_params, void** bufs,
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiled_cpu_function.cc37 bool allocate_entry_params = in XlaCompiledCpuFunction() local
42 /*allocate_entry_params=*/allocate_entry_params, buffer_table_, in XlaCompiledCpuFunction()