| /external/llvm-project/mlir/lib/IR/ |
| D | TypeDetail.h | 211 BaseMemRefTypeStorage(Type elementType, unsigned memorySpace) in BaseMemRefTypeStorage()
212 : ShapedTypeStorage(elementType), memorySpace(memorySpace) {} in BaseMemRefTypeStorage()
215 const unsigned memorySpace; member
221 AffineMap const *affineMapList, const unsigned memorySpace) in MemRefTypeStorage()
222 : BaseMemRefTypeStorage(elementType, memorySpace), in MemRefTypeStorage()
232 return key == KeyTy(getShape(), elementType, getAffineMaps(), memorySpace);
274 UnrankedMemRefTypeStorage(Type elementTy, const unsigned memorySpace) in UnrankedMemRefTypeStorage()
275 : BaseMemRefTypeStorage(elementTy, memorySpace) {} in UnrankedMemRefTypeStorage()
280 return key == KeyTy(elementType, memorySpace);
|
| D | BuiltinTypes.cpp | 384 return static_cast<ImplType *>(impl)->memorySpace; in getMemorySpace()
397 unsigned memorySpace) { in get() argument
398 auto result = getImpl(shape, elementType, affineMapComposition, memorySpace, in get()
412 unsigned memorySpace, Location location) { in getChecked() argument
413 return getImpl(shape, elementType, affineMapComposition, memorySpace, in getChecked()
423 unsigned memorySpace, in getImpl() argument
468 memorySpace); in getImpl()
482 unsigned memorySpace) { in get() argument
483 return Base::get(elementType.getContext(), elementType, memorySpace); in get()
487 unsigned memorySpace, in getChecked() argument
[all …]
|
| /external/llvm-project/mlir/lib/CAPI/IR/ |
| D | BuiltinTypes.cpp | 226 unsigned memorySpace) { in mlirMemRefTypeGet() argument
231 unwrap(elementType), maps, memorySpace)); in mlirMemRefTypeGet()
236 unsigned memorySpace) { in mlirMemRefTypeContiguousGet() argument
239 unwrap(elementType), llvm::None, memorySpace)); in mlirMemRefTypeContiguousGet()
244 unsigned memorySpace, in mlirMemRefTypeContiguousGetChecked() argument
248 llvm::None, memorySpace, unwrap(loc))); in mlirMemRefTypeContiguousGetChecked()
268 MlirType mlirUnrankedMemRefTypeGet(MlirType elementType, unsigned memorySpace) { in mlirUnrankedMemRefTypeGet() argument
269 return wrap(UnrankedMemRefType::get(unwrap(elementType), memorySpace)); in mlirUnrankedMemRefTypeGet()
273 unsigned memorySpace, in mlirUnrankedMemRefTypeGetChecked() argument
275 return wrap(UnrankedMemRefType::getChecked(unwrap(elementType), memorySpace, in mlirUnrankedMemRefTypeGetChecked()
|
| /external/llvm-project/mlir/include/mlir/IR/ |
| D | BuiltinTypes.h | 554 memorySpace(other.getMemorySpace()) {} in Builder()
558 : shape(shape), elementType(elementType), affineMaps(), memorySpace(0) { in Builder()
577 memorySpace = newMemorySpace; in setMemorySpace()
582 return MemRefType::get(shape, elementType, affineMaps, memorySpace); in MemRefType()
589 unsigned memorySpace; variable
600 unsigned memorySpace = 0);
610 unsigned memorySpace, Location location);
631 unsigned memorySpace, Optional<Location> location);
648 static UnrankedMemRefType get(Type elementType, unsigned memorySpace);
654 static UnrankedMemRefType getChecked(Type elementType, unsigned memorySpace,
[all …]
|
| /external/llvm-project/mlir/include/mlir-c/ |
| D | BuiltinTypes.h | 228 MlirAttribute const *affineMaps, unsigned memorySpace);
237 unsigned memorySpace);
243 unsigned memorySpace, MlirLocation loc);
248 unsigned memorySpace);
253 MlirType elementType, unsigned memorySpace, MlirLocation loc);
|
| /external/llvm-project/mlir/lib/Parser/ |
| D | TypeParser.cpp | 226 Optional<unsigned> memorySpace; in parseMemRefType() local
232 if (memorySpace) in parseMemRefType()
234 memorySpace = getToken().getUnsignedIntegerValue(); in parseMemRefType()
235 if (!memorySpace.hasValue()) in parseMemRefType()
242 if (memorySpace) in parseMemRefType()
288 return UnrankedMemRefType::get(elementType, memorySpace.getValueOr(0)); in parseMemRefType()
291 memorySpace.getValueOr(0)); in parseMemRefType()
|
| /external/llvm-project/mlir/include/mlir/Analysis/ |
| D | Utils.h | 306 int memorySpace = -1);
|
| /external/llvm-project/debuginfo-tests/llvm-prettyprinters/gdb/ |
| D | mlir-support.gdb | 39 # CHECK: memorySpace = 6
|
| /external/llvm-project/mlir/lib/Dialect/GPU/IR/ |
| D | GPUDialect.cpp | 722 unsigned memorySpace) { in verifyAttributions() argument
728 if (type.getMemorySpace() != memorySpace) { in verifyAttributions()
730 << "expected memory space " << memorySpace << " in attribution"; in verifyAttributions()
|
| /external/llvm-project/mlir/lib/Analysis/ |
| D | Utils.cpp | 992 int memorySpace) { in getMemoryFootprintBytes() argument
1034 int memorySpace) { in getMemoryFootprintBytes() argument
1038 std::next(Block::iterator(forInst)), memorySpace); in getMemoryFootprintBytes()
|
| /external/llvm-project/mlir/lib/Bindings/Python/ |
| D | IRModules.cpp | 2389 unsigned memorySpace, DefaultingPyLocation loc) { in bindDerived() argument
2391 elementType, shape.size(), shape.data(), memorySpace, loc); in bindDerived()
2434 [](PyType &elementType, unsigned memorySpace, in bindDerived()
2437 mlirUnrankedMemRefTypeGetChecked(elementType, memorySpace, loc); in bindDerived()
|
| /external/llvm-project/mlir/lib/Conversion/StandardToLLVM/ |
| D | StandardToLLVM.cpp | 2651 unsigned memorySpace = in extractPointersAndOffset() local
2657 llvmElementType.getPointerTo(memorySpace).getPointerTo(); in extractPointersAndOffset()
|