Searched refs:adjust_computation (Results 1 – 1 of 1) sorted by relevance
471 auto adjust_computation = in AdjustCalledComputationParameters() local502 adjust_computation(hlo->fused_instructions_computation(), in AdjustCalledComputationParameters()506 adjust_computation(hlo->while_condition(), hlo->operands()); in AdjustCalledComputationParameters()507 adjust_computation(hlo->while_body(), hlo->operands()); in AdjustCalledComputationParameters()511 adjust_computation(hlo->branch_computation(i), in AdjustCalledComputationParameters()521 auto adjust_computation = [this, hlo](HloComputation* computation, in AdjustCalledComputationRoot() local560 adjust_computation(hlo->fused_instructions_computation(), hlo); in AdjustCalledComputationRoot()563 adjust_computation(hlo->while_body(), hlo); in AdjustCalledComputationRoot()567 adjust_computation(branch, hlo); in AdjustCalledComputationRoot()