| /external/llvm-project/mlir/tools/mlir-vulkan-runner/ |
| D | vulkan-runtime-wrappers.cpp | 68 template <typename T, int N> struct MemRefDescriptor { struct
78 BindingIndex bindIndex, MemRefDescriptor<T, S> *ptr) { in bindMemRef() argument
125 BindingIndex bindIndex, MemRefDescriptor<type, size> *ptr) { \
147 _mlir_ciface_fillResource1DFloat(MemRefDescriptor<float, 1> *ptr, // NOLINT in _mlir_ciface_fillResource1DFloat()
154 _mlir_ciface_fillResource2DFloat(MemRefDescriptor<float, 2> *ptr, // NOLINT in _mlir_ciface_fillResource2DFloat()
161 _mlir_ciface_fillResource3DFloat(MemRefDescriptor<float, 3> *ptr, // NOLINT in _mlir_ciface_fillResource3DFloat()
169 _mlir_ciface_fillResource1DInt(MemRefDescriptor<int32_t, 1> *ptr, // NOLINT in _mlir_ciface_fillResource1DInt()
176 _mlir_ciface_fillResource2DInt(MemRefDescriptor<int32_t, 2> *ptr, // NOLINT in _mlir_ciface_fillResource2DInt()
183 _mlir_ciface_fillResource3DInt(MemRefDescriptor<int32_t, 3> *ptr, // NOLINT in _mlir_ciface_fillResource3DInt()
191 _mlir_ciface_fillResource1DInt8(MemRefDescriptor<int8_t, 1> *ptr, // NOLINT in _mlir_ciface_fillResource1DInt8()
[all …]
|
| /external/llvm-project/mlir/lib/Conversion/StandardToLLVM/ |
| D | StandardToLLVM.cpp | 142 return MemRefDescriptor::pack(builder, loc, *this, resultType, inputs); in LLVMTypeConverter()
441 values[i] = MemRefDescriptor::fromStaticShape(rewriter, loc, *this, in promoteBarePtrsToDescriptors()
507 MemRefDescriptor::MemRefDescriptor(Value descriptor) in MemRefDescriptor() function in MemRefDescriptor
515 MemRefDescriptor MemRefDescriptor::undef(OpBuilder &builder, Location loc, in undef()
520 return MemRefDescriptor(descriptor); in undef()
526 MemRefDescriptor
527 MemRefDescriptor::fromStaticShape(OpBuilder &builder, Location loc, in fromStaticShape()
546 auto descr = MemRefDescriptor::undef(builder, loc, convertedType); in fromStaticShape()
560 Value MemRefDescriptor::allocatedPtr(OpBuilder &builder, Location loc) { in allocatedPtr()
565 void MemRefDescriptor::setAllocatedPtr(OpBuilder &builder, Location loc, in setAllocatedPtr()
[all …]
|
| /external/llvm-project/mlir/include/mlir/Conversion/StandardToLLVM/ |
| D | ConvertStandardToLLVM.h | 263 class MemRefDescriptor : public StructBuilder {
266 explicit MemRefDescriptor(Value descriptor);
268 static MemRefDescriptor undef(OpBuilder &builder, Location loc,
273 static MemRefDescriptor fromStaticShape(OpBuilder &builder, Location loc,
556 MemRefDescriptor
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | tf_framework_legalize_to_llvm.cc | 117 MemRefDescriptor memRefDescriptor = CreateMemRefDescriptor( in matchAndRewrite()
144 MemRefDescriptor CreateMemRefDescriptor(Location loc, in CreateMemRefDescriptor()
149 auto memref_desc = MemRefDescriptor::undef( in CreateMemRefDescriptor()
223 MemRefDescriptor memref(transformed.memref()); in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Conversion/GPUCommon/ |
| D | GPUOpsLowering.h | 112 auto descr = MemRefDescriptor::fromStaticShape( in matchAndRewrite()
138 auto descr = MemRefDescriptor::fromStaticShape( in matchAndRewrite()
|
| D | ConvertLaunchFuncToRuntimeCalls.cpp | 399 MemRefDescriptor(adaptor.memref()).allocatedPtr(rewriter, loc); in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Conversion/LinalgToLLVM/ |
| D | LinalgToLLVM.cpp | 97 : d(MemRefDescriptor::undef(rewriter(), loc(), type)) {} in BaseViewConversionHelper()
125 MemRefDescriptor d;
|
| /external/llvm-project/mlir/lib/Conversion/VectorToLLVM/ |
| D | ConvertVectorToLLVM.cpp | 175 base = MemRefDescriptor(memref).alignedPtr(rewriter, loc); in getBase()
186 auto pType = MemRefDescriptor(memref).getElementPtrType(); in getBasePtr()
213 auto pType = MemRefDescriptor(memref).getElementPtrType(); in getIndexedPtrs()
1146 MemRefDescriptor sourceMemRef(operands[0]); in matchAndRewrite()
1169 auto desc = MemRefDescriptor::undef(rewriter, loc, llvmTargetDescriptorTy); in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Conversion/SPIRVToLLVM/ |
| D | ConvertLaunchFuncToLLVMCalls.cpp | 218 MemRefDescriptor descriptor(operand.value()); in matchAndRewrite()
|
| /external/llvm-project/mlir/docs/ |
| D | ConversionToLLVMDialect.md | 410 struct MemRefDescriptor {
|