Searched defs:SetSharding (Results 1 – 2 of 2) sorted by relevance
159 void SetSharding(const OpSharding& sharding) { sharding_ = sharding; } in SetSharding() function1067 void SetSharding(const absl::optional<OpSharding>& sharding) { in SetSharding() function
748 def SetSharding(self, sharding): member in ComputationBuilder