Searched refs:exit_sharding (Results 1 – 1 of 1) sorted by relevance
1928 optional<OpSharding> exit_sharding; in ParseDomain() local1931 attrs["exit"] = {/*required=*/true, AttrTy::kSharding, &exit_sharding}; in ParseDomain()1939 HloSharding::FromProto(*exit_sharding).ValueOrDie()); in ParseDomain()