Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DTransforms.cpp871 bool removeH = (khSize == 1 && ohSize == 1); in returningMatchAndRewrite() local
873 if (!removeH && !removeW) in returningMatchAndRewrite()
880 RTTBuilder(inputType).dropDim((removeH ? 1 : 2)); in returningMatchAndRewrite()
882 RTTBuilder(kernelType).dropDim((removeH ? 0 : 1)); in returningMatchAndRewrite()
884 RTTBuilder(outputType).dropDim(removeH ? 1 : 2); in returningMatchAndRewrite()
898 strides.erase(strides.begin() + (removeH ? 0 : 1)); in returningMatchAndRewrite()
941 bool removeH = (khSize == 1 && ohSize == 1); in returningMatchAndRewrite() local
943 if (!removeH && !removeW) in returningMatchAndRewrite()
950 RTTBuilder(inputType).dropDim((removeH ? 1 : 2)); in returningMatchAndRewrite()
954 RTTBuilder(outputType).dropDim(removeH ? 1 : 2); in returningMatchAndRewrite()
[all …]