Searched refs:ivs_val_flag (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lhlo_legalize_to_parallel_loops.cc | 579 explicit IterArgs(ValueRange ivs_val_flag) : ivs_val_flag_(ivs_val_flag) {} in IterArgs() argument
614 IterArgs ivs_val_flag(window_loops.inner_loop.getRegionIterArgs()); in SelectIvs() local
625 s_and_s_op, mapped_ivs.ivs, &ivs_val_flag, &in_bounds_then_b); in SelectIvs()
633 in_bounds_else_b.create<scf::YieldOp>(loc, ivs_val_flag.to_vector()); in SelectIvs()
642 IterArgs* ivs_val_flag, in SelectOrInitialize() argument
648 TypeRange iter_arg_types{ivs_val_flag->to_vector()}; in SelectOrInitialize()
652 b->create<scf::IfOp>(loc, iter_arg_types, ivs_val_flag->is_init(), in SelectOrInitialize()
662 ApplySingleResultLhloCode(loc, {operand_elem, ivs_val_flag->value()}, in SelectOrInitialize()
679 if_pred_else_b.create<scf::YieldOp>(loc, ivs_val_flag->to_vector()); in SelectOrInitialize()
|