diff options
author | 2018-08-29 10:34:08 -0700 | |
---|---|---|
committer | 2018-08-29 10:38:34 -0700 | |
commit | 150dee25d82589ca109957cc996efbd2a236e044 (patch) | |
tree | ef608bb703027596ed47ea4f30fa6eab93389005 /tensorflow/compiler/xla/service/hlo_instructions.cc | |
parent | aca93368a979419360c1fd84b53b1766b19ba81a (diff) |
[XLA] Implement variadic reduce in the evaluator. It is currently supported only for the case where all of the inputs have the same element type.
PiperOrigin-RevId: 210746149
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_instructions.cc')
-rw-r--r-- | tensorflow/compiler/xla/service/hlo_instructions.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_instructions.cc b/tensorflow/compiler/xla/service/hlo_instructions.cc index ffc74cfedd..0b7f741d73 100644 --- a/tensorflow/compiler/xla/service/hlo_instructions.cc +++ b/tensorflow/compiler/xla/service/hlo_instructions.cc @@ -586,7 +586,7 @@ std::unique_ptr<HloInstruction> HloReduceInstruction::CloneWithNewOperandsImpl( const Shape& shape, tensorflow::gtl::ArraySlice<HloInstruction*> new_operands, HloCloneContext* context) const { - CHECK_EQ(new_operands.size(), 2); + CHECK_EQ(new_operands.size() % 2, 0); return absl::make_unique<HloReduceInstruction>(shape, new_operands, dimensions(), to_apply()); } |