Support sharded ops