diff options
author | HyoukJoong Lee <hyouklee@google.com> | 2018-09-04 13:56:42 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-04 14:00:48 -0700 |
commit | 97039a80b3dabb5ed2e4fb5d0d0bdc5229293718 (patch) | |
tree | d1c318c5863200ac91da19deb100b607c7f64767 /tensorflow/compiler/xla/service/hlo_sharding_metadata.cc | |
parent | ffd9519c3fffe43473f06a1c8fdd12519490db3b (diff) |
Fix CRS combiner for spatial partitioning
PiperOrigin-RevId: 211519250
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_sharding_metadata.cc')
-rw-r--r-- | tensorflow/compiler/xla/service/hlo_sharding_metadata.cc | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_sharding_metadata.cc b/tensorflow/compiler/xla/service/hlo_sharding_metadata.cc index 34cba6136f..e3f4a9852a 100644 --- a/tensorflow/compiler/xla/service/hlo_sharding_metadata.cc +++ b/tensorflow/compiler/xla/service/hlo_sharding_metadata.cc @@ -422,6 +422,13 @@ bool ShardingMetadata::Matches(const DomainMetadata& other) const { : false; } +size_t ShardingMetadata::Hash() const { + if (sharding_ != nullptr) { + return sharding_->Hash(); + } + return static_cast<size_t>(0x297814aaad196e6dULL); +} + string ShardingMetadata::ToString() const { return sharding_ != nullptr ? sharding_->ToString() : "{}"; } |