Searched refs:tupleOp (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | mhlo_control_flow_to_scf.cc | 59 auto tupleOp = whileOp.getOperand().getDefiningOp<TupleOp>(); in MatchAndRewrite() local
60 if (!tupleOp) return; in MatchAndRewrite()
85 return {tupleOp.getOperand(index), index}; in MatchAndRewrite()
114 input.reserve(tupleOp.getNumOperands()); 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()
|
| /external/llvm-project/mlir/lib/Dialect/Vector/ |
| D | VectorTransforms.cpp | 443 Value tupleOp = builder.create<vector::TupleOp>(op->getLoc(), tupleType, in unrollSingleResultStructuredOp() local
452 op->getLoc(), resultVectorType, tupleOp, builder.getI64ArrayAttr(sizes), in unrollSingleResultStructuredOp()
623 Value tupleOp = in unrollTransferReadOp() local
627 loc, sourceVectorType, tupleOp, builder.getI64ArrayAttr(targetShape), in unrollTransferReadOp()
748 auto tupleOp = dyn_cast_or_null<vector::TupleOp>( in matchAndRewrite() local
750 if (!tupleOp) in matchAndRewrite()
771 loc, tupleOp.getOperand(index), xferWriteOp.memref(), sliceIndices, in matchAndRewrite()
895 } else if (auto tupleOp = dyn_cast<vector::TupleOp>(op)) { in getProducerValue() local
899 auto value = tupleOp.getOperand(tupleIndex); in getProducerValue()
|
| D | VectorOps.cpp | 2941 if (auto tupleOp = getOperand().getDefiningOp<TupleOp>()) in fold() local
2942 return tupleOp.getOperand(getIndex()); in fold()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 610 if (auto tupleOp = in fold() local
612 return tupleOp.getOperand(index()); in fold()
|