From de4c12857782f65dc4a941776d506ecac50a5934 Mon Sep 17 00:00:00 2001 From: Michael Kuperstein Date: Thu, 2 Aug 2018 12:46:13 -0700 Subject: [XLA] Introduce variadic version of reduce. This defines the semantics, and adds parser and shape inference support. Since support is not plumbed through the rest of the compiler here, multi-output reduce is still rejected by the HLO verifier, and is not exposed through XlaBuilder. PiperOrigin-RevId: 207148035 --- tensorflow/compiler/xla/service/hlo_instructions.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'tensorflow/compiler/xla/service/hlo_instructions.h') diff --git a/tensorflow/compiler/xla/service/hlo_instructions.h b/tensorflow/compiler/xla/service/hlo_instructions.h index 3797bef600..ac5a1ca080 100644 --- a/tensorflow/compiler/xla/service/hlo_instructions.h +++ b/tensorflow/compiler/xla/service/hlo_instructions.h @@ -331,7 +331,7 @@ class HloConcatenateInstruction : public HloInstruction { class HloReduceInstruction : public HloInstruction { public: explicit HloReduceInstruction( - const Shape& shape, HloInstruction* arg, HloInstruction* init_value, + const Shape& shape, tensorflow::gtl::ArraySlice args, tensorflow::gtl::ArraySlice dimensions_to_reduce, HloComputation* reduce_computation); // Returns the dimension sizes or numbers associated with this instruction. -- cgit v1.2.3