Searched defs:kernelSpatialDimensions (Results 1 – 5 of 5) sorted by relevance
248 MlirContext ctx) { in PYBIND11_MODULE()
250 intptr_t nKernelSpatialDimensions, const int64_t *kernelSpatialDimensions, in mlirMhloConvDimensionNumbersGet()
1879 auto kernelSpatialDimensions = in isSpatialDimensionsValid() local2118 auto kernelSpatialDimensions = in verify() local6080 SmallVector<int64_t> kernelSpatialDimensions; in parseConvolutionDimensionsRaw() local6280 llvm::SmallVector<int64_t> kernelSpatialDimensions = parsedDims.first; in parseConvolutionDimensions() local
2174 auto kernelSpatialDimensions = in isSpatialDimensionsValid() local2517 auto kernelSpatialDimensions = in verify() local8906 SmallVector<int64_t> kernelSpatialDimensions; in parseConvolutionDimensionsRaw() local9106 llvm::SmallVector<int64_t> kernelSpatialDimensions = parsedDims.first; in parseConvolutionDimensions() local
2002 auto kernelSpatialDimensions = in matchAndRewrite() local