Home
last modified time | relevance | path

Searched defs:memrefType (Results 1 – 25 of 46) sorted by relevance

12

/llvm-project/mlir/lib/Dialect/MLProgram/Transforms/
H A DBufferizableOpInterfaceImpl.cpp61 auto memrefType = getMemRefTypeWithStaticIdentityLayout(tensorType); in bufferize() local
98 auto memrefType = getMemRefTypeWithStaticIdentityLayout(tensorType); in bufferize() local
128 auto memrefType = getMemRefTypeWithStaticIdentityLayout(tensorType); in bufferize() local
/llvm-project/mlir/lib/Conversion/LinalgToStandard/
H A DLinalgToStandard.cpp45 if (auto memrefType = dyn_cast<MemRefType>(type)) in extractOperandTypes() local
99 auto memrefType = dyn_cast<MemRefType>(op.getType()); in createTypeCanonicalizedMemRefOperands() local
/llvm-project/mlir/lib/Conversion/BufferizationToMemRef/
H A DBufferizationToMemRef.cpp77 MemRefType memrefType = MemRefType::get({ShapedType::kDynamic}, in matchAndRewrite() local
86 MemRefType memrefType = cast<MemRefType>(type); in matchAndRewrite() local
[all...]
/llvm-project/mlir/lib/Dialect/MemRef/IR/
H A DMemRefMemorySlot.cpp114 MemRefType memrefType = getType(); in getDestructurableSlots() local
136 auto memrefType = llvm::cast<DestructurableTypeInterface>(getType()); in destructure() local
200 MemRefType memrefType) { in getAttributeIndexFromIndexOperands()
308 auto memrefType = llvm::cast<MemRefType>(type); in getSubelementIndexMap() local
325 auto memrefType = llvm::cast<MemRefType>(type); in getTypeAtIndex() local
H A DValueBoundsOpInterfaceImpl.cpp82 auto memrefType = llvm::dyn_cast<MemRefType>(rankOp.getMemref().getType()); populateBoundsForIndexValue() local
H A DMemRefOps.cpp69 auto memrefType = llvm::cast<MemRefType>(value.getType()); in getMixedSize() local
79 auto memrefType = llvm::cast<MemRefType>(value.getType()); in getMixedSizes() local
163 getConstantOffset(MemRefType memrefType) getConstantOffset() argument
176 getConstantStrides(MemRefType memrefType) getConstantStrides() argument
253 auto memrefType = alloc.getType(); matchAndRewrite() local
1025 auto memrefType = llvm::dyn_cast<MemRefType>(getSource().getType()); fold() local
1372 auto memrefType = inferReturnTypes() local
1477 if (auto memrefType = llvm::dyn_cast<MemRefType>(memref.getType())) { build() local
1511 Type memrefType; parse() local
1574 auto memrefType = llvm::dyn_cast<MemRefType>(type); parseGlobalMemrefOpTypeAndInitialValue() local
1598 auto memrefType = llvm::dyn_cast<MemRefType>(getType()); verify() local
2926 auto memrefType = llvm::cast<ShapedType>(expectedType); produceSubViewErrorMsg() local
3080 auto memrefType = llvm::cast<MemRefType>(memref.getType()); createCanonicalRankReducingSubViewOp() local
3457 auto memrefType = viewOp.getType(); matchAndRewrite() local
[all...]
/llvm-project/mlir/lib/Dialect/MemRef/Transforms/
H A DNormalizeMemRefs.cpp229 MemRefType memrefType = dyn_cast<MemRefType>(opType); updateFunctionSignature() local
361 MemRefType memrefType = dyn_cast<MemRefType>(argType); normalizeFuncOpMemRefs() local
468 MemRefType memrefType = dyn_cast<MemRefType>(resType); normalizeFuncOpMemRefs() local
508 MemRefType memrefType = dyn_cast<MemRefType>(resultType); createOpResultsNormalized() local
[all...]
H A DExpandStridedMetadata.cpp946 auto memrefType = cast<MemRefType>(reinterpretCastOp.getResult().getType()); matchAndRewrite() local
1020 auto memrefType = cast<MemRefType>(source.getType()); matchAndRewrite() local
/llvm-project/mlir/test/lib/Analysis/
H A DTestMemRefStrideCalculation.cpp35 auto memrefType = cast<MemRefType>(allocOp.getResult().getType()); in runOnOperation() local
/llvm-project/mlir/lib/Conversion/GPUToSPIRV/
H A DWmmaOpsToSPIRV.cpp217 MemRefType memrefType = op.getSrcMemref().getType(); in matchAndRewrite() local
254 auto memrefType = cast<MemRefType>(op.getDstMemref().getType()); in matchAndRewrite() local
/llvm-project/mlir/include/mlir/Dialect/Bufferization/IR/
H A DUnstructuredControlFlow.h63 if (auto memrefType = in getBufferType() local
/llvm-project/mlir/lib/Dialect/Bufferization/Transforms/
H A DBufferResultsToOutParams.cpp60 if (auto memrefType = dyn_cast<MemRefType>(resultType.value())) { updateFuncOp() local
173 auto memrefType = cast<MemRefType>(memref.getType()); updateCalls() local
[all...]
H A DBufferUtils.cpp142 auto memrefType = cast<MemRefType>(typeConverter.convertType(type)); getGlobalFor() local
H A DFuncBufferizableOpInterfaceImpl.cpp68 BaseMemRefType memrefType = options.functionArgTypeConverterFn( getBufferizedFunctionArgType() local
/llvm-project/mlir/lib/Conversion/MemRefToSPIRV/
H A DMemRefToSPIRV.cpp386 auto memrefType = cast<MemRefType>(atomicOp.getMemref().getType()); in matchAndRewrite() local
516 auto memrefType = cast<MemRefType>(loadOp.getMemref().getType()); in matchAndRewrite() local
633 auto memrefType = cast<MemRefType>(loadOp.getMemref().getType()); in matchAndRewrite() local
657 auto memrefType = cast<MemRefType>(storeOp.getMemref().getType()); matchAndRewrite() local
857 auto memrefType = cast<MemRefType>(storeOp.getMemref().getType()); matchAndRewrite() local
[all...]
/llvm-project/mlir/lib/Dialect/Arith/Transforms/
H A DBufferizableOpInterfaceImpl.cpp197 auto memrefType = llvm::cast<MemRefType>(*trueType); in getBufferType() local
/llvm-project/mlir/lib/Dialect/NVGPU/Transforms/
H A DCreateAsyncGroups.cpp127 static bool resultsInSupportedAsyncCopy(MemRefType memrefType, in resultsInSupportedAsyncCopy() argument
/llvm-project/mlir/unittests/IR/
H A DShapedTypeTest.cpp32 ShapedType memrefType = TEST() local
/llvm-project/mlir/lib/Dialect/Affine/IR/
H A DAffineOps.cpp2983 auto memrefType = llvm::cast<MemRefType>(operands[0].getType()); build() local
2992 auto memrefType = llvm::cast<MemRefType>(memref.getType()); build() local
2999 auto memrefType = llvm::cast<MemRefType>(memref.getType()); build() local
3044 verifyMemoryOpIndexing(Operation * op,AffineMapAttr mapAttr,Operation::operand_range mapOperands,MemRefType memrefType,unsigned numIndexOperands) verifyMemoryOpIndexing() argument
3064 auto memrefType = getMemRefType(); verify() local
3136 auto memrefType = llvm::cast<MemRefType>(memref.getType()); build() local
3180 auto memrefType = getMemRefType(); verify() local
4346 auto memrefType = llvm::cast<MemRefType>(memref.getType()); build() local
4365 MemRefType memrefType; parse() local
4395 verifyVectorMemoryOp(Operation * op,MemRefType memrefType,VectorType vectorType) verifyVectorMemoryOp() argument
4405 MemRefType memrefType = getMemRefType(); verify() local
4437 auto memrefType = llvm::cast<MemRefType>(memref.getType()); build() local
4454 MemRefType memrefType; parse() local
4487 MemRefType memrefType = getMemRefType(); verify() local
[all...]
/llvm-project/mlir/lib/Dialect/Tensor/Transforms/
H A DBufferizableOpInterfaceImpl.cpp494 if (failed(memrefType)) in bufferize() local
213 auto memrefType = bufferize() local
893 auto memrefType = MemRefType::get( bufferize() local
/llvm-project/mlir/lib/Conversion/VectorToSPIRV/
H A DVectorToSPIRV.cpp722 spirv::getElementPtr(typeConverter, memrefType, adaptor.getBase(), in matchAndRewrite() local
687 auto memrefType = loadOp.getMemRefType(); matchAndRewrite() local
/llvm-project/mlir/lib/Conversion/GPUCommon/
H A DGPUOpsLowering.cpp566 getDynamicSharedMemorySymbol(ConversionPatternRewriter & rewriter,Operation * moduleOp,gpu::DynamicSharedMemoryOp op,const LLVMTypeConverter * typeConverter,MemRefType memrefType,unsigned alignmentBit) getDynamicSharedMemorySymbol() argument
619 MemRefType memrefType = op.getResultMemref().getType(); matchAndRewrite() local
/llvm-project/mlir/lib/Dialect/Vector/Utils/
H A DVectorUtils.cpp257 bool vector::isContiguousSlice(MemRefType memrefType, VectorType vectorType) { in isContiguousSlice() argument
/llvm-project/mlir/lib/Dialect/NVGPU/IR/
H A DNVGPUDialect.cpp343 verifyTmaDescriptorWithMemref(Operation * op,nvgpu::TensorMapDescriptorType descType,std::optional<MemRefType> memrefType=std::nullopt) verifyTmaDescriptorWithMemref() argument
/llvm-project/mlir/lib/Dialect/Bufferization/IR/
H A DBufferizableOpInterface.cpp630 ensureToMemrefOpIsValid(Value tensor,Type memrefType) ensureToMemrefOpIsValid() argument
653 FailureOr<BaseMemRefType> memrefType = getBufferType(value, options); getBuffer() local

12