Searched refs:ShardableValueUpdatePair (Results 1 – 2 of 2) sorted by relevance
60 struct ShardableValueUpdatePair { struct189 const std::vector<ShardableValueUpdatePair> shardable_value_update_pairs() in shardable_value_update_pairs()194 std::vector<ShardableValueUpdatePair> pairs) { in set_shardable_value_update_pairs()279 std::vector<ShardableValueUpdatePair> shardable_value_update_pairs_;
61 const std::vector<xla::HloModuleConfig::ShardableValueUpdatePair>66 std::vector<xla::HloModuleConfig::ShardableValueUpdatePair> pairs) { in set_shardable_value_update_pairs()83 std::vector<xla::HloModuleConfig::ShardableValueUpdatePair>