Home
last modified time | relevance | path

Searched defs:allocOp (Results 1 – 9 of 9) sorted by relevance

/external/llvm-project/mlir/test/lib/Transforms/
DTestMemRefStrideCalculation.cpp26 getFunction().walk([&](AllocOp allocOp) { in runOnFunction()
/external/llvm-project/mlir/lib/Transforms/
DNormalizeMemRefs.cpp155 .walk([&](AllocOp allocOp) -> WalkResult { in areMemRefsNormalizable()
332 for (AllocOp allocOp : allocOps) in normalizeFuncOpMemRefs() local
DPipelineDataTransfer.cpp277 if (auto *allocOp = oldMemRef.getDefiningOp()) { in runOnAffineForOp() local
DBufferOptimizations.cpp321 Operation *allocOp = alloc.getDefiningOp(); in promote() local
/external/llvm-project/mlir/lib/Transforms/Utils/
DUtils.cpp396 LogicalResult mlir::normalizeMemRef(AllocOp allocOp) { in normalizeMemRef()
/external/tensorflow/tensorflow/compiler/xla/service/mlir_gpu/
Dpasses.cc119 mlir::Operation* allocOp = SearchAllocOp(memref); in GetAllocOp() local
/external/llvm-project/mlir/lib/Conversion/GPUCommon/
DConvertLaunchFuncToRuntimeCalls.cpp346 gpu::AllocOp allocOp, ArrayRef<Value> operands, in matchAndRewrite()
/external/llvm-project/mlir/lib/Conversion/StandardToLLVM/
DStandardToLLVM.cpp2035 AllocOp allocOp = cast<AllocOp>(op); in allocateBuffer() local
2128 AllocOp allocOp = cast<AllocOp>(op); in allocateBuffer() local
/external/llvm-project/mlir/lib/Dialect/StandardOps/IR/
DOps.cpp4313 AllocOp allocOp = allocOperand.getDefiningOp<AllocOp>(); in matchAndRewrite() local