Home
last modified time | relevance | path

Searched refs:indexedGenericOp (Results 1 – 4 of 4) sorted by relevance

/external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
DLoops.cpp462 IndexedGenericOp indexedGenericOp) { in emitScalarImplementation() argument
463 assert(indexedGenericOp.hasBufferSemantics() && in emitScalarImplementation()
467 unsigned nInputs = indexedGenericOp.getNumInputs(); in emitScalarImplementation()
468 unsigned nOutputs = indexedGenericOp.getNumOutputs(); in emitScalarImplementation()
480 b, loc, indexedGenericOp.getInputIndexingMap(i), allIvs); in emitScalarImplementation()
483 IndexedValueType(indexedGenericOp.getInput(i))(indexing)); in emitScalarImplementation()
488 b, loc, indexedGenericOp.getOutputIndexingMap(i), allIvs); in emitScalarImplementation()
491 IndexedValueType(indexedGenericOp.getOutputBuffer(i))(indexing)); in emitScalarImplementation()
501 b, loc, indexedGenericOp.getOutputIndexingMap(i), allIvs)); in emitScalarImplementation()
502 outputBuffers.push_back(indexedGenericOp.getOutputBuffer(i)); in emitScalarImplementation()
[all …]
DTiling.cpp167 auto indexedGenericOp = dyn_cast<IndexedGenericOp>(op.getOperation()); in transformIndexedGenericOpIndices() local
168 if (!indexedGenericOp) in transformIndexedGenericOpIndices()
177 auto &region = indexedGenericOp.region(); in transformIndexedGenericOpIndices()
179 indexedGenericOp.emitOpError("expected a region"); in transformIndexedGenericOpIndices()
186 for (unsigned i = 0; i < indexedGenericOp.getNumLoops(); ++i) { in transformIndexedGenericOpIndices()
193 Value newIndex = b.create<AddIOp>(indexedGenericOp.getLoc(), oldIndex, in transformIndexedGenericOpIndices()
DFusion.cpp138 if (auto indexedGenericOp = dyn_cast<IndexedGenericOp>(clonedOp)) { in cloneWithLoopRanges() local
139 auto &block = indexedGenericOp.region().front(); in cloneWithLoopRanges()
142 for (unsigned i = 0, e = indexedGenericOp.getNumLoops(); i < e; ++i) { in cloneWithLoopRanges()
145 AddIOp newIndex = b.create<AddIOp>(indexedGenericOp.getLoc(), oldIndex, in cloneWithLoopRanges()
/external/llvm-project/mlir/lib/Conversion/LinalgToStandard/
DLinalgToStandard.cpp28 if (auto indexedGenericOp = dyn_cast<IndexedGenericOp>(op)) { in extractOperandTypes() local
30 auto numLoops = indexedGenericOp.getNumLoops(); in extractOperandTypes()
172 auto indexedGenericOp = cast<IndexedGenericOp>(op); in matchAndRewrite() local
173 auto numLoops = indexedGenericOp.getNumLoops(); in matchAndRewrite()