Searched refs:host_shape (Results 1 – 3 of 3) sorted by relevance
341 xla::DeviceMemoryAllocator* allocator, xla::Shape* host_shape, in ExpandTreeOfTuples() argument345 *host_shape = elements.shape(); in ExpandTreeOfTuples()364 *xla::ShapeUtil::GetMutableSubshape(host_shape, index) = in ExpandTreeOfTuples()388 xla::Shape host_shape; in MakeTuple() local391 &host_shape, &device_shape)); in MakeTuple()397 host_shape, device_shape); in MakeTuple()
213 xla::DeviceMemoryAllocator* allocator, xla::Shape* host_shape,
51 virtual Shape HostShapeToDeviceShape(const Shape& host_shape) const { in HostShapeToDeviceShape() argument52 return host_shape; in HostShapeToDeviceShape()