Searched refs:entry_sharding (Results 1 – 1 of 1) sorted by relevance
1927 optional<OpSharding> entry_sharding; in ParseDomain() local1930 attrs["entry"] = {/*required=*/true, AttrTy::kSharding, &entry_sharding}; in ParseDomain()1937 HloSharding::FromProto(*entry_sharding).ValueOrDie()); in ParseDomain()