Searched defs:inputBatchDimension (Results 1 – 5 of 5) sorted by relevance
248 MlirContext ctx) { in PYBIND11_MODULE()
247 MlirContext ctx, int64_t inputBatchDimension, int64_t inputFeatureDimension, in mlirMhloConvDimensionNumbersGet()
6075 int64_t inputBatchDimension = 0; in parseConvolutionDimensionsRaw() local6274 int64_t inputBatchDimension = parsedDims.second[IOBatch]; in parseConvolutionDimensions() local
1994 auto inputBatchDimension = dimensionNumbers.getInputBatchDimension(); in matchAndRewrite() local
8901 int64_t inputBatchDimension = 0; in parseConvolutionDimensionsRaw() local9100 int64_t inputBatchDimension = parsedDims.second[IOBatch]; in parseConvolutionDimensions() local