Searched defs:HloSharding (Results 1 – 2 of 2) sorted by relevance
357 explicit HloSharding(bool manual, bool replicated, in HloSharding() function372 explicit HloSharding(int64_t device_id, absl::Span<const OpMetadata> metadata) in HloSharding() function401 explicit HloSharding(const std::vector<HloSharding>& tuple_shardings) in HloSharding() function
397 HloSharding = _xla.HloSharding variable