Searched refs:root_allocation (Results 1 – 3 of 3) sorted by relevance
202 const se::DeviceMemoryBase& root_allocation() const;
358 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()
198 const BufferAllocation* root_allocation = in EmitComputation() local200 if (root_allocation && root_allocation->is_thread_local()) { in EmitComputation()