Searched refs:root_allocation (Results 1 – 3 of 3) sorted by relevance
209 const se::DeviceMemoryBase& root_allocation() const;
359 const se::DeviceMemoryBase& XRTTupleAllocation::root_allocation() const { in root_allocation() function in tensorflow::XRTTupleAllocation521 tuple_buffers.set_buffer(element.allocation->root_allocation(), index); in MakeTuple()
223 const BufferAllocation* root_allocation = in EmitComputation() local225 if (root_allocation && in EmitComputation()226 (root_allocation->is_thread_local() || in EmitComputation()227 (root_allocation->is_constant() && has_thread_local_param))) { in EmitComputation()