Searched defs:set_sharding (Results 1 – 1 of 1) sorted by relevance
1158 void set_sharding(const HloSharding& sharding) { in set_sharding() function1161 void set_sharding(std::shared_ptr<const HloSharding> sharding) { in set_sharding() function