Searched defs:outputFeatureDimension (Results 1 – 5 of 5) sorted by relevance
248 MlirContext ctx) { in PYBIND11_MODULE()
251 int64_t outputBatchDimension, int64_t outputFeatureDimension, in mlirMhloConvDimensionNumbersGet()
6082 int64_t outputFeatureDimension = 0; in parseConvolutionDimensionsRaw() local6291 const int64_t outputFeatureDimension = parsedDims.second[IOFeature]; in parseConvolutionDimensions() local
2005 auto outputFeatureDimension = dimensionNumbers.getOutputFeatureDimension(); in matchAndRewrite() local
8908 int64_t outputFeatureDimension = 0; in parseConvolutionDimensionsRaw() local9117 const int64_t outputFeatureDimension = parsedDims.second[IOFeature]; in parseConvolutionDimensions() local