Home
last modified time | relevance | path

Searched refs:set_tuple_index (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dconditional_simplifier.cc115 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()
H A Dhlo_instruction.h2002 void set_tuple_index(int64_t new_tuple_index);
H A Dconditional_code_motion.cc811 user->set_tuple_index(user->tuple_index() - 1); in MoveInstructionIn()
H A Dhlo_instructions.h1453 void set_tuple_index(int64_t new_tuple_index) { in set_tuple_index() function
H A Dhlo_instruction.cc4663 void HloInstruction::set_tuple_index(int64_t new_tuple_index) { in set_tuple_index() function in xla::HloInstruction
4664 return Cast<HloGetTupleElementInstruction>(this)->set_tuple_index( in set_tuple_index()
H A Dhlo_instructions.cc2274 proto.set_tuple_index(tuple_index_); in ToProto()
H A Dalgebraic_simplifier.cc3397 use->set_tuple_index(index_map[use->tuple_index()]); in HandleOptimizationBarrier()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dxla_builder.cc1353 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()