Home
last modified time | relevance | path

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

/external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
DFusionOnTensors.cpp188 SmallVector<Value, 2> fusedOperands; in fuseTensorOpsImpl() local
189 fusedOperands.reserve(numFusedOperands); in fuseTensorOpsImpl()
192 fusedOperands.assign(consumerOperands.begin(), in fuseTensorOpsImpl()
194 fusedOperands.append(producerOperands.begin(), producerOperands.end()); in fuseTensorOpsImpl()
195 fusedOperands.append(std::next(consumerOperands.begin(), consumerIdx + 1), in fuseTensorOpsImpl()
204 fusedIndexMaps.reserve(fusedOperands.size() + consumer.getNumOutputs()); in fuseTensorOpsImpl()
223 /*inputs=*/fusedOperands, in fuseTensorOpsImpl()
236 /*inputs=*/fusedOperands, in fuseTensorOpsImpl()
705 SmallVector<Value, 2> fusedOperands(linalgOp.getInputs()); in matchAndRewrite() local
706 fusedOperands[operand.index()] = reshapeOp.src(); in matchAndRewrite()
[all …]