Lines Matching refs:Tuple
53 // A Tuple constructed of a bunch of parameters should not be changed.
69 // A GTE of a tuple parameter should not be changed.
82 // A GTE of a Tuple should be short-circuited.
90 HloInstruction* tuple = builder.AddInstruction(
93 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple, 1));
106 // Verify a chain of GTE/Tuple instructions is collapsed.
114 HloInstruction* tuple = builder.AddInstruction(
117 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple, 1));
126 op::Negate(op::GetTupleElement(op::Tuple())));
134 // Verify a nesting of GTE/Tuple instructions is collapsed. Tuples are nested
135 // to some depth with a chain of Tuple instructions, then extracted with a
165 // Verify that a tuple constructed of GTE instructions operating on the same
166 // tuple are collapsed.
176 HloInstruction* tuple =
182 EXPECT_THAT(computation->root_instruction(), tuple);
190 // Verify that a tuple->GTE->tuple construct is not simplified if the input
191 // and output tuple are not compatible shapes.
199 // Output tuple has only two elements. Parameter tuple has three elements so
201 HloInstruction* tuple =
207 EXPECT_THAT(computation->root_instruction(), tuple);
211 EXPECT_THAT(computation->root_instruction(), tuple);
