Searched refs:root_allocation (Results 1 – 2 of 2) sorted by relevance
169 const se::DeviceMemoryBase& root_allocation();
268 const se::DeviceMemoryBase& XRTTupleAllocation::root_allocation() { in root_allocation() function in tensorflow::XRTTupleAllocation447 tuple_buffers.set_buffer(element.allocation->root_allocation(), index); in MakeTuple()