Searched defs:while_body_root (Results 1 – 5 of 5) sorted by relevance
50 auto* while_body_root = while_body_comp->root_instruction(); in RunWhileDCE() local
59 HloInstruction* while_body_root = while_body->root_instruction(); in TryRemoveDeadWhileParams() local316 auto* while_body_root = while_body->root_instruction(); in TryRemoveConstantParams() local513 auto while_body_root = while_body->root_instruction(); in TryPropagateConstant() local648 auto* while_body_root = while_body->root_instruction(); in TryFlattenNestedTuples() local775 auto* while_body_root = while_body->root_instruction(); in TryMergeInductionVariables() local
129 auto* while_body_root = while_body->root_instruction(); in GetLoopInductionVarTupleIdx() local449 auto* while_body_root = while_body->root_instruction(); in ComputeWhileLoopTripCountUpperBound() local
30 HloInstruction* while_body_root, int64 tuple_index) { in ReplaceUsesWhileKeepingLoopInvariance()
55 auto* while_body_root = while_body_comp->root_instruction(); in WhileBodyHasPassThroughTupleElement() local