Searched refs:contractingDimMap (Results 1 – 2 of 2) sorted by relevance
288 const std::vector<std::pair<int64_t, int64_t>> &contractingDimMap, in verifyOutputShape() argument292 for (auto &dimPair : contractingDimMap) { in verifyOutputShape()406 auto contractingDimMap = op.getContractingDimMap(); in verify() local410 if (contractingDimMap.empty()) in verify()414 if (!verifyDimMap(lhsType, rhsType, contractingDimMap)) in verify()423 contractingDimMap, batchDimMap))) in verify()
1967 std::vector<std::pair<int64_t, int64_t>> contractingDimMap = in matchAndRewrite() local1971 for (auto &dimPair : contractingDimMap) { in matchAndRewrite()1997 if (!contractingDimMap.empty()) { in matchAndRewrite()