Searched defs:user_sharding (Results 1 – 1 of 1) sorted by relevance
739 std::optional<HloSharding> user_sharding = in InferShardingFromUsers() local757 std::optional<HloSharding> user_sharding = in InferShardingFromUsers() local1109 std::optional<HloSharding> user_sharding = in InferUnspecifiedDimsFromOneUser() local1517 auto user_sharding = in GetShardingFromUser() local1549 HloSharding user_sharding = user.sharding(); in GetShardingFromUser() local