Searched refs:windowDims (Results 1 – 3 of 3) sorted by relevance
98 ArrayRef<AffineExpr> windowDims);
409 The window dimensions are specified by argument `windowDims`. The i-th410 dimension in the shape of `windowDims` denotes the size of the window along412 should be passed to the operation as `windowDims`.416 AnyStridedMemRef:$windowDims,441 auto windowDims = makeAffineDimExprs(nWin, idx, context);444 weightedPoolingInputIndex(*this, outputDims, windowDims);448 // windowDims449 AffineMap::get(idx, 0, windowDims, context),
1266 auto windowDimsType = op.windowDims().getType().template cast<MemRefType>(); in verifySingleInputPoolingOp()1364 ArrayRef<AffineExpr> windowDims) { in weightedPoolingInputIndex() argument1365 assert(outputDims.size() == windowDims.size()); in weightedPoolingInputIndex()1371 op.getDilation(i) * windowDims[i] - op.getLowPad(i); in weightedPoolingInputIndex()1381 ArrayRef<AffineExpr> windowDims);