Searched refs:scoped_sharding (Results 1 – 1 of 1) sorted by relevance
1316 XlaScopedShardingAssignment scoped_sharding(this, sharding); in Infeed() local1335 XlaScopedShardingAssignment scoped_sharding(this, in Infeed() local1428 XlaScopedShardingAssignment scoped_sharding(this, OpSharding()); in Outfeed() local