diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-07-15 16:04:03 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-15 16:07:40 -0700 |
| commit | 6c3c766dcabff3b5fa41dbfd491c9e8062a77b07 (patch) | |
| tree | 744fa2bc46e9b446e5645bd4b6962ce641898fd6 /tensorflow/compiler/xla/service/hlo_instruction.h | |
| parent | e5945c00148186808e337b4946cf0fa6460f6803 (diff) | |
[XLA] Enable the semantic for cross-modeul AllReduce.
PiperOrigin-RevId: 204670087
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_instruction.h')
| -rw-r--r-- | tensorflow/compiler/xla/service/hlo_instruction.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_instruction.h b/tensorflow/compiler/xla/service/hlo_instruction.h index cbd78fa124..180b2fb359 100644 --- a/tensorflow/compiler/xla/service/hlo_instruction.h +++ b/tensorflow/compiler/xla/service/hlo_instruction.h @@ -1128,6 +1128,9 @@ class HloInstruction { // Returns true if this instruction is elementwise on all its operands. bool IsElementwise() const; + // Returns true if this is an cross module all-reduce instrucion. + bool IsCrossModuleAllReduce() const; + // Returns true if this elementwise instruction implicitly broadcasts operand // `operand_idx`. // |