Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_state.h209 const se::DeviceMemoryBase& root_allocation() const;
Dxrt_state.cc359 const se::DeviceMemoryBase& XRTTupleAllocation::root_allocation() const { in root_allocation() function in tensorflow::XRTTupleAllocation
521 tuple_buffers.set_buffer(element.allocation->root_allocation(), index); in MakeTuple()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc223 const BufferAllocation* root_allocation = in EmitComputation() local
225 if (root_allocation && in EmitComputation()
226 (root_allocation->is_thread_local() || in EmitComputation()
227 (root_allocation->is_constant() && has_thread_local_param))) { in EmitComputation()