Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/
H A DLinalgInterfaces.cpp318 llvm::SmallDenseSet<unsigned> allLoopDims; in isConvolutionInterfaceImpl() local
327 allLoopDims.insert(outputDim); in isConvolutionInterfaceImpl()
336 allLoopDims.insert(outputDim); in isConvolutionInterfaceImpl()
346 allLoopDims.insert(outputDim); in isConvolutionInterfaceImpl()
355 allLoopDims.insert(outputDim); in isConvolutionInterfaceImpl()
374 if (allLoopDims.count(filterDim)) in isConvolutionInterfaceImpl()
376 allLoopDims.insert(filterDim); in isConvolutionInterfaceImpl()
385 if (allLoopDims.count(filterDim)) in isConvolutionInterfaceImpl()
387 allLoopDims.insert(filterDim); in isConvolutionInterfaceImpl()
398 if (allLoopDims.size() != linalgOp.getNumLoops()) in isConvolutionInterfaceImpl()