Searched defs:host_shape (Results 1 – 2 of 2) sorted by relevance
51 virtual Shape HostShapeToDeviceShape(const Shape& host_shape) const { in HostShapeToDeviceShape()
341 xla::DeviceMemoryAllocator* allocator, xla::Shape* host_shape, in ExpandTreeOfTuples()388 xla::Shape host_shape; in MakeTuple() local