Searched defs:while_body_root (Results 1 – 5 of 5) sorted by relevance
65 HloInstruction* while_body_root = while_body->root_instruction(); in RemoveDeadTupleIndices() local240 HloInstruction* while_body_root = while_body->root_instruction(); in TryRemoveDeadWhileParams() local538 HloInstruction* while_body_root = while_body->root_instruction(); in TryRemoveRepeatedWhileTupleIndices() local634 auto* while_body_root = while_body->root_instruction(); in TryRemoveConstantParams() local868 auto while_body_root = while_body->root_instruction(); in TryPropagateConstant() local1003 auto* while_body_root = while_body->root_instruction(); in TryFlattenNestedTuples() local1134 auto* while_body_root = while_body->root_instruction(); in TryMergeInductionVariables() local
146 auto* while_body_root = while_body->root_instruction(); in GetAuxiliaryLoopInductionVars() local282 auto* while_body_root = while_body->root_instruction(); in GetLoopInductionVarTupleIdx() local603 auto* while_body_root = while_body->root_instruction(); in ComputeWhileLoopTripCountUpperBound() local
54 auto* while_body_root = while_body_comp->root_instruction(); in RunWhileDCE() local
32 HloInstruction* while_body_root, in ReplaceUsesWhileKeepingLoopInvariance()
53 auto* while_body_root = while_body_comp->root_instruction(); in WhileBodyHasPassThroughTupleElement() local