Searched refs:AddGetTupleElements (Results 1 – 5 of 5) sorted by relevance
152 TEST_F(TupleUtilTest, AddGetTupleElements) { in TEST_F() argument168 HloInstruction* new_gte = TupleUtil::AddGetTupleElements({p0, {1, 0}}); in TEST_F()
61 static HloInstruction* AddGetTupleElements(const HloPosition& position);
146 /*static*/ HloInstruction* TupleUtil::AddGetTupleElements( in AddGetTupleElements() function in xla::TupleUtil
3601 HloInstruction* producing_instruction = AddGetTupleElements(); in Process()3626 HloInstruction* MemorySpaceAssignment::Allocation::AddGetTupleElements() const { in AddGetTupleElements() function in xla::memory_space_assignment::MemorySpaceAssignment::Allocation3633 return TupleUtil::AddGetTupleElements(defining_position()); in AddGetTupleElements()3675 HloInstruction* producing_instruction = AddGetTupleElements(); in Process()3725 original_allocation_.AddGetTupleElements(); in Process()3763 AddGetTupleElements(), while_body->root_instruction(), in PostProcess()
612 HloInstruction* AddGetTupleElements() const;