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