Searched defs:kernelOutputFeatureDimension (Results 1 – 5 of 5) sorted by relevance
248 MlirContext ctx) { in PYBIND11_MODULE()
249 int64_t kernelInputFeatureDimension, int64_t kernelOutputFeatureDimension, in mlirMhloConvDimensionNumbersGet()
2000 auto kernelOutputFeatureDimension = in matchAndRewrite() local2336 int64_t kernelOutputFeatureDimension = in matchAndRewrite() local
6079 int64_t kernelOutputFeatureDimension = 0; in parseConvolutionDimensionsRaw() local6282 int64_t kernelOutputFeatureDimension = parsedDims.second[KOFeature]; in parseConvolutionDimensions() local
8905 int64_t kernelOutputFeatureDimension = 0; in parseConvolutionDimensionsRaw() local9108 int64_t kernelOutputFeatureDimension = parsedDims.second[KOFeature]; in parseConvolutionDimensions() local