Searched refs:induction_vars (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lhlo_legalize_to_affine.cc | 119 ValueRange induction_vars) { in matchAndRewrite() argument
120 auto l = builder.create<AffineLoadOp>(loc, lhs, induction_vars); in matchAndRewrite()
121 auto r = builder.create<AffineLoadOp>(loc, rhs, induction_vars); in matchAndRewrite()
126 rewriter.create<AffineStoreOp>(loc, op_result, op.out(), induction_vars); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_loop_simplifier.cc | 1002 absl::flat_hash_map<int64, const HloConstantInstruction*> induction_vars; in TryMergeInductionVariables() local
1018 induction_vars.emplace(i, Cast<HloConstantInstruction>(constant)); in TryMergeInductionVariables()
1029 if (induction_vars.size() + (trip_counter.has_value() ? 1 : 0) < 2) { in TryMergeInductionVariables()
1083 if (!induction_vars.count(i)) { in TryMergeInductionVariables()
1091 add_new_instr(induction_vars.at(i)->Clone())))); in TryMergeInductionVariables()
1109 add_gte((induction_vars.count(i) ? loop_body_param : old_root), i)); in TryMergeInductionVariables()
|