Searched refs:scoped_sharding (Results 1 – 1 of 1) sorted by relevance
1374 XlaScopedShardingAssignment scoped_sharding(this, sharding); in Infeed() local1392 XlaScopedShardingAssignment scoped_sharding(this, in Infeed() local1484 XlaScopedShardingAssignment scoped_sharding(this, OpSharding()); in Outfeed() local