Searched refs:AsOwningDeviceMemory (Results 1 – 4 of 4) sorted by relevance
44 MaybeOwningDeviceMemory::AsOwningDeviceMemory() const { in AsOwningDeviceMemory() function in xla::MaybeOwningDeviceMemory
62 const tensorflow::se::OwningDeviceMemory* AsOwningDeviceMemory() const;
68 index_buffer.second.AsOwningDeviceMemory(); in ToShapedBuffer()
50 mem.AsOwningDeviceMemory(); in ToC()