Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dbufferize.cc83 rewriter.replaceOpWithNewOp<DimOp>(op, adaptor.memrefOrTensor(), in matchAndRewrite()
96 rewriter.replaceOpWithNewOp<RankOp>(op, adaptor.memrefOrTensor()); in matchAndRewrite()
Dsame_shape_propagation.cc293 if (!candidate) candidate = dimOp.memrefOrTensor(); in tryEvaluateShapeToRoot()
296 return candidate == dimOp.memrefOrTensor() && in tryEvaluateShapeToRoot()
/external/llvm-project/mlir/lib/Dialect/StandardOps/Transforms/
DBufferize.cpp31 rewriter.replaceOpWithNewOp<DimOp>(op, adaptor.memrefOrTensor(), in matchAndRewrite()
/external/llvm-project/mlir/lib/Dialect/StandardOps/IR/
DOps.cpp1329 Value memrefOrTensor, int64_t index) { in build() argument
1332 build(builder, result, memrefOrTensor, indexValue); in build()
1336 Value memrefOrTensor, Value index) { in build() argument
1338 build(builder, result, indexTy, memrefOrTensor, index); in build()
1354 auto type = op.memrefOrTensor().getType(); in verify()
1377 auto argTy = memrefOrTensor().getType(); in fold()
1390 Operation *definingOp = memrefOrTensor().getDefiningOp(); in fold()
1452 auto reshape = dim.memrefOrTensor().getDefiningOp<MemRefReshapeOp>(); in matchAndRewrite()
/external/llvm-project/mlir/lib/Conversion/StandardToLLVM/
DStandardToLLVM.cpp2918 Type operandType = dimOp.memrefOrTensor().getType(); in matchAndRewrite()
2948 UnrankedMemRefDescriptor unrankedDesc(transformed.memrefOrTensor()); in extractSizeOfUnrankedMemRef()
2987 MemRefDescriptor descriptor(transformed.memrefOrTensor()); in extractSizeOfRankedMemRef()
2996 MemRefDescriptor memrefDescriptor(transformed.memrefOrTensor()); in extractSizeOfRankedMemRef()
3008 Type operandType = op.memrefOrTensor().getType(); in matchAndRewrite()
3010 UnrankedMemRefDescriptor desc(RankOp::Adaptor(operands).memrefOrTensor()); in matchAndRewrite()
/external/llvm-project/mlir/lib/Dialect/Affine/IR/
DAffineOps.cpp175 return isTopLevelValue(dimOp.memrefOrTensor()); in isValidDim()
199 if (isTopLevelValue(dimOp.memrefOrTensor())) in isDimOpValidSymbol()
204 if (dimOp.memrefOrTensor().isa<BlockArgument>()) in isDimOpValidSymbol()
213 return TypeSwitch<Operation *, bool>(dimOp.memrefOrTensor().getDefiningOp()) in isDimOpValidSymbol()
/external/llvm-project/mlir/include/mlir/Dialect/StandardOps/IR/
DOps.td1561 "any tensor or memref type">:$memrefOrTensor,
1566 attr-dict $memrefOrTensor `,` $index `:` type($memrefOrTensor)
1570 OpBuilderDAG<(ins "Value":$memrefOrTensor, "int64_t":$index)>,
1571 OpBuilderDAG<(ins "Value":$memrefOrTensor, "Value":$index)>
2467 "any tensor or memref type">:$memrefOrTensor);
2478 let assemblyFormat = "$memrefOrTensor attr-dict `:` type($memrefOrTensor)";