Searched refs:ToDeviceMemoryTree (Results 1 – 2 of 2) sorted by relevance
189 xla::ShapeTree<xla::MaybeOwningDeviceMemory> ToDeviceMemoryTree(
541 XRTTupleAllocation::ToDeviceMemoryTree( in ToDeviceMemoryTree() function in tensorflow::XRTTupleAllocation