Searched defs:while_body_root (Results 1 – 5 of 5) sorted by relevance
70 HloInstruction* while_body_root = while_body->root_instruction(); in RemoveDeadTupleIndices() local244 HloInstruction* while_body_root = while_body->root_instruction(); in TryRemoveDeadWhileParams() local434 HloInstruction* while_body_root = while_body->root_instruction(); in TryRemoveRepeatedWhileTupleIndices() local530 auto* while_body_root = while_body->root_instruction(); in TryRemoveConstantParams() local763 auto while_body_root = while_body->root_instruction(); in TryPropagateConstant() local898 auto* while_body_root = while_body->root_instruction(); in TryFlattenNestedTuples() local1028 auto* while_body_root = while_body->root_instruction(); in TryMergeInductionVariables() local
139 auto* while_body_root = while_body->root_instruction(); in GetAuxiliaryLoopInductionVars() local275 auto* while_body_root = while_body->root_instruction(); in GetLoopInductionVarTupleIdx() local595 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_t tuple_index) { in ReplaceUsesWhileKeepingLoopInvariance()
54 auto* while_body_root = while_body_comp->root_instruction(); in WhileBodyHasPassThroughTupleElement() local