Searched defs:while_body_root (Results 1 – 5 of 5) sorted by relevance
55 HloInstruction* while_body_root = while_body->root_instruction(); in RemoveDeadTupleIndices() local227 HloInstruction* while_body_root = while_body->root_instruction(); in TryRemoveDeadWhileParams() local401 HloInstruction* while_body_root = while_body->root_instruction(); in TryRemoveRepeatedWhileTupleIndices() local497 auto* while_body_root = while_body->root_instruction(); in TryRemoveConstantParams() local725 auto while_body_root = while_body->root_instruction(); in TryPropagateConstant() local860 auto* while_body_root = while_body->root_instruction(); in TryFlattenNestedTuples() local987 auto* while_body_root = while_body->root_instruction(); in TryMergeInductionVariables() local
135 auto* while_body_root = while_body->root_instruction(); in GetAuxiliaryLoopInductionVars() local271 auto* while_body_root = while_body->root_instruction(); in GetLoopInductionVarTupleIdx() local591 auto* while_body_root = while_body->root_instruction(); in ComputeWhileLoopTripCountUpperBound() local
53 auto* while_body_root = while_body_comp->root_instruction(); in RunWhileDCE() local
30 HloInstruction* while_body_root, int64 tuple_index) { in ReplaceUsesWhileKeepingLoopInvariance()
54 auto* while_body_root = while_body_comp->root_instruction(); in WhileBodyHasPassThroughTupleElement() local