Searched refs:dimOp (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | same_shape_propagation.cc | 291 auto dimOp = val.value().getDefiningOp<DimOp>(); in tryEvaluateShapeToRoot() local
292 if (!dimOp) return false; in tryEvaluateShapeToRoot()
293 if (!candidate) candidate = dimOp.memrefOrTensor(); in tryEvaluateShapeToRoot()
294 auto index = dimOp.getConstantIndex(); in tryEvaluateShapeToRoot()
296 return candidate == dimOp.memrefOrTensor() && in tryEvaluateShapeToRoot()
|
| /external/llvm-project/mlir/lib/Dialect/Affine/IR/ |
| D | AffineOps.cpp | 174 if (auto dimOp = dyn_cast<DimOp>(op)) in isValidDim() local
175 return isTopLevelValue(dimOp.memrefOrTensor()); in isValidDim()
196 static bool isDimOpValidSymbol(DimOp dimOp, Region *region) { in isDimOpValidSymbol() argument
199 if (isTopLevelValue(dimOp.memrefOrTensor())) in isDimOpValidSymbol()
204 if (dimOp.memrefOrTensor().isa<BlockArgument>()) in isDimOpValidSymbol()
209 Optional<int64_t> index = dimOp.getConstantIndex(); in isDimOpValidSymbol()
213 return TypeSwitch<Operation *, bool>(dimOp.memrefOrTensor().getDefiningOp()) in isDimOpValidSymbol()
282 if (auto dimOp = dyn_cast<DimOp>(defOp)) in isValidSymbol() local
283 return isDimOpValidSymbol(dimOp, region); in isValidSymbol()
|
| /external/llvm-project/mlir/lib/Conversion/StandardToLLVM/ |
| D | StandardToLLVM.cpp | 2916 matchAndRewrite(DimOp dimOp, ArrayRef<Value> operands, in matchAndRewrite()
2918 Type operandType = dimOp.memrefOrTensor().getType(); in matchAndRewrite()
2920 rewriter.replaceOp(dimOp, {extractSizeOfUnrankedMemRef( in matchAndRewrite()
2921 operandType, dimOp, operands, rewriter)}); in matchAndRewrite()
2926 rewriter.replaceOp(dimOp, {extractSizeOfRankedMemRef( in matchAndRewrite()
2927 operandType, dimOp, operands, rewriter)}); in matchAndRewrite()
2934 Value extractSizeOfUnrankedMemRef(Type operandType, DimOp dimOp, in extractSizeOfUnrankedMemRef()
2937 Location loc = dimOp.getLoc(); in extractSizeOfUnrankedMemRef()
2976 Value extractSizeOfRankedMemRef(Type operandType, DimOp dimOp, in extractSizeOfRankedMemRef()
2979 Location loc = dimOp.getLoc(); in extractSizeOfRankedMemRef()
[all …]
|
| /external/llvm-project/mlir/docs/Rationale/ |
| D | UsageOfConst.md | 228 if (auto *dimOp = inst->dyn_cast<DimOp>()) {
|