Searched defs:HloSharding (Results 1 – 1 of 1) sorted by relevance
292 explicit HloSharding(bool manual, bool replicated, in HloSharding() function307 explicit HloSharding(int64 device_id, absl::Span<const OpMetadata> metadata) in HloSharding() function325 explicit HloSharding(const std::vector<HloSharding>& tuple_shardings) in HloSharding() function