Searched refs:element_0 (Results 1 – 2 of 2) sorted by relevance
1883 HloInstruction* element_0 = builder.AddInstruction( in MakeBenchmarkWhileBody() local1896 HloInstruction::CreateTuple({element_0, rev_1, add_1_2})); in MakeBenchmarkWhileBody()1972 HloInstruction* element_0 = builder.AddInstruction( in BM_ParallelWhiles() local1975 x->shape(), HloOpcode::kAdd, sum, element_0)); in BM_ParallelWhiles()
374 auto element_0 = GetTupleElement(param, 0); in XLA_TEST_F() local376 Tuple(&builder, {Neg(element_0), Add(element_1, element_1)}); in XLA_TEST_F()