Searched defs:rhsBatchingDims (Results 1 – 5 of 5) sorted by relevance
97 rhsContractingDims, rhsBatchingDims; in matchAndRewrite() local
1469 auto rhsBatchingDims = dimNumbers.getRhsBatchingDimensions(); in matchAndRewrite() local3160 auto rhsBatchingDims = dimNumbers.getRhsBatchingDimensions(); in matchAndRewrite() local
192 const std::vector<int64_t> &rhsContractingDims, MlirContext ctx) { in PYBIND11_MODULE()
1049 ArrayRef<int64_t> rhsBatchingDims = dimNumbers.getRhsBatchingDimensions(); in verify() local
1047 ArrayRef<int64_t> rhsBatchingDims = dimNumbers.getRhsBatchingDimensions(); in verify() local