Searched defs:outputSpatialDimensions (Results 1 – 5 of 5) sorted by relevance
248 MlirContext ctx) { in PYBIND11_MODULE()
252 intptr_t nOutputSpatialDimensions, const int64_t *outputSpatialDimensions) { in mlirMhloConvDimensionNumbersGet()
1881 auto outputSpatialDimensions = in isSpatialDimensionsValid() local6083 SmallVector<int64_t> outputSpatialDimensions; in parseConvolutionDimensionsRaw() local6289 llvm::SmallVector<int64_t> outputSpatialDimensions = parsedDims.first; in parseConvolutionDimensions() local
2176 auto outputSpatialDimensions = in isSpatialDimensionsValid() local8909 SmallVector<int64_t> outputSpatialDimensions; in parseConvolutionDimensionsRaw() local9115 llvm::SmallVector<int64_t> outputSpatialDimensions = parsedDims.first; in parseConvolutionDimensions() local
2006 auto outputSpatialDimensions = in matchAndRewrite() local