Searched refs:set_tuple_index (Results 1 – 8 of 8) sorted by relevance
115 user->set_tuple_index(map[user->tuple_index()]); in TryRemoveUnusedConditionalOperands()309 user->set_tuple_index(new_index); in RemoveUnusedTupleElements()414 user->set_tuple_index(i); in MergeDuplicateTupleElements()
2002 void set_tuple_index(int64_t new_tuple_index);
811 user->set_tuple_index(user->tuple_index() - 1); in MoveInstructionIn()
1453 void set_tuple_index(int64_t new_tuple_index) { in set_tuple_index() function
4663 void HloInstruction::set_tuple_index(int64_t new_tuple_index) { in set_tuple_index() function in xla::HloInstruction4664 return Cast<HloGetTupleElementInstruction>(this)->set_tuple_index( in set_tuple_index()
2274 proto.set_tuple_index(tuple_index_); in ToProto()
3397 use->set_tuple_index(index_map[use->tuple_index()]); in HandleOptimizationBarrier()
1353 instr.set_tuple_index(index); in GetTupleElementInternal()1815 infeed_data.set_tuple_index(0); in Infeed()3434 recv_data.set_tuple_index(0); in Recv()