Searched refs:adjust_computation (Results 1 – 1 of 1) sorted by relevance
408 auto adjust_computation = in AdjustCalledComputationParameters() local439 adjust_computation(hlo->fused_instructions_computation(), in AdjustCalledComputationParameters()443 adjust_computation(hlo->while_condition(), hlo->operands()); in AdjustCalledComputationParameters()444 adjust_computation(hlo->while_body(), hlo->operands()); in AdjustCalledComputationParameters()452 auto adjust_computation = [this, hlo](HloComputation* computation, in AdjustCalledComputationRoot() local491 adjust_computation(hlo->fused_instructions_computation(), hlo); in AdjustCalledComputationRoot()494 adjust_computation(hlo->while_body(), hlo); in AdjustCalledComputationRoot()