Searched refs:adjust_computation (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | bfloat16_propagation.cc | 462 auto adjust_computation = [this, hlo]( in AdjustCalledComputationParameters() local
492 adjust_computation(hlo->fused_instructions_computation(), in AdjustCalledComputationParameters()
496 adjust_computation(hlo->while_condition(), hlo->operands()); in AdjustCalledComputationParameters()
497 adjust_computation(hlo->while_body(), hlo->operands()); in AdjustCalledComputationParameters()
501 adjust_computation(hlo->branch_computation(i), in AdjustCalledComputationParameters()
511 auto adjust_computation = [this, hlo](HloComputation* computation, in AdjustCalledComputationRoot() local
550 adjust_computation(hlo->fused_instructions_computation(), hlo); in AdjustCalledComputationRoot()
553 adjust_computation(hlo->while_body(), hlo); in AdjustCalledComputationRoot()
557 adjust_computation(branch, hlo); in AdjustCalledComputationRoot()
|