Category: dtensor.call_with_layout