Searched refs:outputDims (Results 1 – 3 of 3) sorted by relevance
| /external/llvm-project/mlir/include/mlir/Dialect/Linalg/IR/ |
| D | LinalgOps.h | 97 weightedPoolingInputIndex(PoolingOp op, ArrayRef<AffineExpr> outputDims,
|
| D | LinalgStructuredOps.td | 440 auto outputDims = makeAffineDimExprs(nPar, idx, context);
444 weightedPoolingInputIndex(*this, outputDims, windowDims);
451 AffineMap::get(idx, 0, outputDims, context)});
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/IR/ |
| D | LinalgOps.cpp | 1363 ArrayRef<AffineExpr> outputDims, in weightedPoolingInputIndex() argument
1365 assert(outputDims.size() == windowDims.size()); in weightedPoolingInputIndex()
1367 res.reserve(outputDims.size()); in weightedPoolingInputIndex()
1368 for (unsigned i = 0, e = outputDims.size(); i < e; ++i) { in weightedPoolingInputIndex()
1370 auto expr = op.getStride(i) * outputDims[i] + in weightedPoolingInputIndex()
1380 OP_TYPE op, ArrayRef<AffineExpr> outputDims, \
|