- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for IsReplicatedSharding (0.24 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.h
// type but not replicated. bool IsSplitSharding(const xla::OpSharding& sharding); // Returns whether the sharding is replicated. It includes sharding with // REPLICATED type and replicated OTHER type. bool IsReplicatedSharding(const xla::OpSharding& sharding); // Returns a map of dimension indices and number of splits for tiled sharding. absl::StatusOr<std::map<int, int>> GetDimensionIndicesAndNumSplitsFromSharding(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 28 22:18:34 UTC 2024 - 6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.cc
} bool IsSplitSharding(const xla::OpSharding& sharding) { return sharding.type() == xla::OpSharding::OTHER && !IsOtherReplicatedSharding(sharding); } bool IsReplicatedSharding(const xla::OpSharding& sharding) { return sharding.type() == xla::OpSharding::REPLICATED || IsOtherReplicatedSharding(sharding); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 21:28:13 UTC 2024 - 34K bytes - Viewed (0)