Searched refs:HostShapeToDeviceShape (Results 1 – 5 of 5) sorted by relevance
158 const Shape on_device_shape = HostShapeToDeviceShape(literal.shape()); in TransferArrayToDeviceAsync()181 if (!ShapeUtil::Equal(HostShapeToDeviceShape(shape), shape)) { in TransferArrayFromDevice()184 ShapeUtil::HumanString(HostShapeToDeviceShape(shape))); in TransferArrayFromDevice()317 const Shape on_device_shape = HostShapeToDeviceShape(on_host_shape); in AllocateScopedShapedBuffer()
51 virtual Shape HostShapeToDeviceShape(const Shape& host_shape) const { in HostShapeToDeviceShape() function
51 client->backend().transfer_manager()->HostShapeToDeviceShape( in AllocateShapedBuffer()
239 client_->backend().transfer_manager()->HostShapeToDeviceShape(shape); in PopulateInputs()
101 xla::Shape on_device_shape = transfer_manager->HostShapeToDeviceShape(shape); in AllocateScopedShapedBuffer()