Home
last modified time | relevance | path

Searched defs:while_input (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dwhile_loop_simplifier.cc356 HloInstruction* while_input = comp->parameter_instruction(0); in TryRemoveDeadWhileParams() local
367 HloInstruction* while_input = comp->parameter_instruction(0); in TryRemoveDeadWhileParams() local
422 HloInstruction* while_input = comp->parameter_instruction(0); in TryRemoveDeadWhileParams() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Ddtensor_allreduce_sum_optimization.cc316 mlir::OpOperand& while_input = while_op->getOpOperand(block_arg_index); in CheckWhileLoopOptimizationCriteria() local