Searched defs:kernelInputFeatureDimension (Results 1 – 5 of 5) sorted by relevance
248 MlirContext ctx) { in PYBIND11_MODULE()
249 int64_t kernelInputFeatureDimension, int64_t kernelOutputFeatureDimension, in mlirMhloConvDimensionNumbersGet()
1998 auto kernelInputFeatureDimension = in matchAndRewrite() local2334 int64_t kernelInputFeatureDimension = in matchAndRewrite() local
6078 int64_t kernelInputFeatureDimension = 0; in parseConvolutionDimensionsRaw() local6281 int64_t kernelInputFeatureDimension = parsedDims.second[KIFeature]; in parseConvolutionDimensions() local
8904 int64_t kernelInputFeatureDimension = 0; in parseConvolutionDimensionsRaw() local9107 int64_t kernelInputFeatureDimension = parsedDims.second[KIFeature]; in parseConvolutionDimensions() local