Searched refs:batchDimMap (Results 1 – 2 of 2) sorted by relevance
1803 std::vector<std::pair<int64_t, int64_t>> batchDimMap = op.getBatchDimMap(); in matchAndRewrite() local1804 if (!batchDimMap.empty()) { in matchAndRewrite()1805 int64_t lhsIndex = batchDimMap[0].first; in matchAndRewrite()1806 int64_t rhsIndex = batchDimMap[0].second; in matchAndRewrite()
638 const std::vector<std::pair<int64_t, int64_t>> &batchDimMap) { in verifyOutputShape() argument646 for (auto &dimPair : batchDimMap) in verifyOutputShape()760 auto batchDimMap = getBatchDimMap(); in verify() local771 if (!verifyDimMap(lhsType, rhsType, batchDimMap)) in verify()776 contractingDimMap, batchDimMap))) in verify()