Home
last modified time | relevance | path

Searched refs:loopIndVar (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dmhlo_control_flow_to_scf.cc91 ValueIndex loopIndVar = getValueAndIndex(cmp.lhs()); in MatchAndRewrite() local
93 if (!loopIndVar.first || !max.first) return; in MatchAndRewrite()
95 bodyReturn.getOperand(loopIndVar.second).getDefiningOp<mhlo::AddOp>(); in MatchAndRewrite()
107 llvm::dbgs() << " loopIndVar = " << loopIndVar.second << " max = " in MatchAndRewrite()
109 llvm::dbgs() << " loopIndVar = " << loopIndVar.first << " max = " in MatchAndRewrite()
115 for (auto r : tupleOp.getOperands().take_front(loopIndVar.second)) in MatchAndRewrite()
117 for (auto r : tupleOp.getOperands().drop_front(loopIndVar.second + 1)) in MatchAndRewrite()
128 auto forloopIndVar = getAsIndex(loopIndVar.first); in MatchAndRewrite()
140 loopIndVar.first.getType().cast<ShapedType>().getElementType(); in MatchAndRewrite()
153 if (gte.index() == loopIndVar.second) { in MatchAndRewrite()
[all …]