Searched refs:memRefDescriptor (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | tf_framework_legalize_to_llvm.cc | 117 MemRefDescriptor memRefDescriptor = CreateMemRefDescriptor( in matchAndRewrite() local
121 rewriter.replaceOp(op, {memRefDescriptor}); in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Conversion/StandardToLLVM/ |
| D | StandardToLLVM.cpp | 1058 MemRefDescriptor memRefDescriptor(memRefDesc); in getStridedElementPtr() local
1059 Value base = memRefDescriptor.alignedPtr(rewriter, loc); in getStridedElementPtr()
1064 ? memRefDescriptor.offset(rewriter, loc) in getStridedElementPtr()
1071 ? memRefDescriptor.stride(rewriter, loc, i) in getStridedElementPtr()
1079 LLVM::LLVMType elementPtrType = memRefDescriptor.getElementPtrType(); in getStridedElementPtr()
1187 auto memRefDescriptor = MemRefDescriptor::undef(rewriter, loc, structType); in createMemRefDescriptor() local
1190 memRefDescriptor.setAllocatedPtr(rewriter, loc, allocatedPtr); in createMemRefDescriptor()
1193 memRefDescriptor.setAlignedPtr(rewriter, loc, alignedPtr); in createMemRefDescriptor()
1196 memRefDescriptor.setOffset(rewriter, loc, in createMemRefDescriptor()
1201 memRefDescriptor.setSize(rewriter, loc, en.index(), en.value()); in createMemRefDescriptor()
[all …]
|
| /external/llvm-project/mlir/lib/Conversion/GPUCommon/ |
| D | ConvertLaunchFuncToRuntimeCalls.cpp | 379 auto memRefDescriptor = this->createMemRefDescriptor( in matchAndRewrite() local
382 rewriter.replaceOp(allocOp, {memRefDescriptor, stream}); in matchAndRewrite()
|