Searched refs:XlaScopedShardingAssignment (Results 1 – 4 of 4) sorted by relevance
109 xla::XlaScopedShardingAssignment assign_sharding(b, op_sharding); in Compute()
210 xla::XlaScopedShardingAssignment assign_sharding( in BuildComputation()283 xla::XlaScopedShardingAssignment assign_sharding( in BuildComputation()800 xla::XlaScopedShardingAssignment assign_tuple_sharding(builder, in BuildArguments()821 xla::XlaScopedShardingAssignment assign_sharding( in BuildArguments()830 xla::XlaScopedShardingAssignment assign_sharding( in BuildArguments()
1011 class XlaScopedShardingAssignment {1013 XlaScopedShardingAssignment(xla::XlaBuilder* builder, in XlaScopedShardingAssignment() function1019 XlaScopedShardingAssignment(const XlaScopedShardingAssignment&) = delete;1020 XlaScopedShardingAssignment& operator=(const XlaScopedShardingAssignment&) =1023 ~XlaScopedShardingAssignment() { SetSharding(prev_sharding_); } in ~XlaScopedShardingAssignment()
1316 XlaScopedShardingAssignment scoped_sharding(this, sharding); in Infeed()1335 XlaScopedShardingAssignment scoped_sharding(this, in Infeed()1428 XlaScopedShardingAssignment scoped_sharding(this, OpSharding()); in Outfeed()