Searched refs:SetSharding (Results 1 – 1 of 1) sorted by relevance
159 void SetSharding(const OpSharding& sharding) { sharding_ = sharding; } in SetSharding() function1016 SetSharding(sharding); in XlaScopedShardingAssignment()1023 ~XlaScopedShardingAssignment() { SetSharding(prev_sharding_); } in ~XlaScopedShardingAssignment()1026 void SetSharding(const absl::optional<OpSharding>& sharding) { in SetSharding() function1028 builder_->SetSharding(sharding.value()); in SetSharding()