Home
last modified time | relevance | path

Searched defs:while_init (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dwhile_loop_simplifier.cc62 HloInstruction* while_init = while_op->mutable_operand(0); in RemoveDeadTupleIndices() local
237 HloInstruction* while_init = while_op->mutable_operand(0); in TryRemoveDeadWhileParams() local
476 HloInstruction* while_init = while_op->mutable_operand(0); in TryRemoveRepeatedWhileTupleIndicesHelper() local
535 HloInstruction* while_init = while_op->mutable_operand(0); in TryRemoveRepeatedWhileTupleIndices() local
631 auto* while_init = while_op->mutable_operand(0); in TryRemoveConstantParams() local
862 auto while_init = while_op->operand(0); in TryPropagateConstant() local
1000 auto* while_init = while_op->mutable_operand(0); in TryFlattenNestedTuples() local
1131 auto* while_init = while_op->mutable_operand(0); in TryMergeInductionVariables() local
H A Dwhile_loop_analysis.cc309 auto* while_init = while_op->operand(0); in GetLoopInductionVarTupleIdx() local
500 auto* while_init = while_op->mutable_operand(0); in ComputeWhileLoopTripCount() local
H A Dwhile_loop_all_reduce_code_motion.cc555 HloInstruction* while_init{nullptr}; member
H A Dheap_simulator_test.cc167 HloInstruction* while_init = in TEST_F() local
H A Dcopy_insertion.cc309 HloInstruction* while_init = xla_while->mutable_operand(0); in AddCopiesForWhile() local
H A Dcopy_insertion_test.cc1802 auto while_init = builder.AddInstruction(HloInstruction::CreateTuple( in TEST_F() local
H A Dhlo_instruction.cc2855 HloInstruction* HloInstruction::while_init() const { in while_init() function in xla::HloInstruction