Searched defs:inputSpatialDimensions (Results 1 – 5 of 5) sorted by relevance
248 MlirContext ctx) { in PYBIND11_MODULE()
248 intptr_t nInputSpatialDimensions, const int64_t *inputSpatialDimensions, in mlirMhloConvDimensionNumbersGet()
1877 auto inputSpatialDimensions = in isSpatialDimensionsValid() local6077 SmallVector<int64_t> inputSpatialDimensions; in parseConvolutionDimensionsRaw() local6273 llvm::SmallVector<int64_t> inputSpatialDimensions = parsedDims.first; in parseConvolutionDimensions() local
2172 auto inputSpatialDimensions = in isSpatialDimensionsValid() local8903 SmallVector<int64_t> inputSpatialDimensions; in parseConvolutionDimensionsRaw() local9099 llvm::SmallVector<int64_t> inputSpatialDimensions = parsedDims.first; in parseConvolutionDimensions() local
1996 auto inputSpatialDimensions = dimensionNumbers.getInputSpatialDimensions(); in matchAndRewrite() local