diff options
| author |  Tong Shen <endlessroad@google.com> | 2018-08-21 10:25:52 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-21 10:33:51 -0700 |
| commit | 938a3b77797164db736a1006a7656326240baa59 (patch) | |
| tree | 73c2697cf3b7c8900694c5c3358f55096f0cdb82 /tensorflow/compiler/xla/service/hlo_module_group_metadata.cc | |
| parent | 792a933b113aa772b5ff5dbb6ef1892ffeb99063 (diff) | |
In HostCompute op, use SendToHost/RecvFromHost instead of Send/Recv.
PiperOrigin-RevId: 209617148
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_module_group_metadata.cc')
| -rw-r--r-- | tensorflow/compiler/xla/service/hlo_module_group_metadata.cc | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_module_group_metadata.cc b/tensorflow/compiler/xla/service/hlo_module_group_metadata.cc index 3b512bf0f8..cd10913763 100644 --- a/tensorflow/compiler/xla/service/hlo_module_group_metadata.cc +++ b/tensorflow/compiler/xla/service/hlo_module_group_metadata.cc @@ -204,6 +204,10 @@ const HloModuleGroupMetadata::Channel& HloModuleGroupMetadata::GetChannel( return channels_[channel_id_map_.at(channel_id)]; } +bool HloModuleGroupMetadata::HasChannel(int64 channel_id) const { + return channel_id_map_.find(channel_id) != channel_id_map_.end(); +} + HloComputation* HloModuleGroupMetadata::PeerComputation( const HloInstruction* instruction) const { CHECK(IsChannelInstruction(instruction)); |