Searched defs:shardings (Results 1 – 4 of 4) sorted by relevance
72 OpSharding Tuple(const ShapeTree<OpSharding>& shardings) { in Tuple()
346 const std::vector<HloSharding>& shardings) { in set_spmd_parameters_shardings()
170 absl::Span<const HloSharding> shardings) { in Tuple()
427 llvm::ArrayRef<absl::optional<xla::OpSharding>> shardings) { in AllOptionalShardingsAreSet()