Home
last modified time | relevance | path

Searched defs:shardings (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/
Dsharding_builder.cc72 OpSharding Tuple(const ShapeTree<OpSharding>& shardings) { in Tuple()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module.h341 const std::vector<HloSharding>& shardings) { in set_spmd_parameters_shardings()
Dhlo_sharding.cc126 absl::Span<const HloSharding> shardings) { in Tuple()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc403 llvm::ArrayRef<absl::optional<xla::OpSharding>> shardings) { in AllOptionalShardingsAreSet()