Searched refs:tensorMemrefType (Results 1 – 6 of 6) sorted by relevance
| /llvm-project/mlir/lib/CAPI/Dialect/ |
| H A D | NVGPU.cpp | 24 MlirType tensorMemrefType, in mlirNVGPUTensorMapDescriptorTypeGet() argument
28 unwrap(ctx), cast<MemRefType>(unwrap(tensorMemrefType)), in mlirNVGPUTensorMapDescriptorTypeGet()
|
| /llvm-project/mlir/lib/Bindings/Python/ |
| H A D | DialectNVGPU.cpp | 26 [](nb::object cls, MlirType tensorMemrefType, int swizzle, int l2promo, in populateDialectNVGPUSubmodule()
29 ctx, tensorMemrefType, swizzle, l2promo, oobFill, interleave)); in populateDialectNVGPUSubmodule()
27 __anonc1027f100102(py::object cls, MlirType tensorMemrefType, int swizzle, int l2promo, int oobFill, int interleave, MlirContext ctx) populateDialectNVGPUSubmodule() argument
|
| /llvm-project/mlir/test/python/dialects/ |
| H A D | nvgpu.py | 21 tensorMemrefType = MemRefType.get(
26 tensorMemrefType,
|
| /llvm-project/mlir/include/mlir-c/Dialect/ |
| H A D | NVGPU.h | 29 MlirContext ctx, MlirType tensorMemrefType, int swizzle, int l2promo,
|
| /llvm-project/mlir/test/Examples/NVGPU/tools/ |
| H A D | nvdsl.py | 125 tensorMemrefType = ir.MemRefType.get(
131 tensorMemrefType,
|
| /llvm-project/mlir/test/Integration/GPU/CUDA/sm90/python/tools/ |
| H A D | matmulBuilder.py | 45 tensorMemrefType = ir.MemRefType.get(
51 tensorMemrefType,
|