Searched refs:SetHostShape (Results 1 – 1 of 1) sorted by relevance
64 SetHostShape(ShapeUtil::DeviceShapeToHostShape(buffers_.shape())); in ExecutionInput()69 SetHostShape(ShapeUtil::DeviceShapeToHostShape(buffers_.shape())); in ExecutionInput()74 SetHostShape(ShapeUtil::DeviceShapeToHostShape(buffers_.shape())); in ExecutionInput()80 SetHostShape(ShapeUtil::DeviceShapeToHostShape(buffers_.shape())); in ExecutionInput()132 void SetHostShape(xla::Shape host_shape) { in SetHostShape() function