Searched defs:opToMask (Results 1 – 1 of 1) sorted by relevance
357 getOrCreateMaskFor(RewriterBase & rewriter,Operation * opToMask,LinalgOp linalgOp,std::optional<AffineMap> maybeMaskingMap) getOrCreateMaskFor() argument 426 maskOperation(RewriterBase & rewriter,Operation * opToMask,LinalgOp linalgOp,std::optional<AffineMap> maybeMaskingMap) maskOperation() argument 3231 __anon729377362e02(ArrayRef<int64_t> maskShape, ArrayRef<bool> scalableDims, Operation *opToMask) depthwiseConv() argument [all...]