Searched refs:scoped_sharding (Results 1 – 1 of 1) sorted by relevance
1697 XlaScopedShardingAssignment scoped_sharding(this, sharding); in Infeed() local1715 XlaScopedShardingAssignment scoped_sharding(this, in Infeed() local1811 XlaScopedShardingAssignment scoped_sharding(this, OpSharding()); in Outfeed() local