Searched refs:batchDimMap (Results 1 – 2 of 2) sorted by relevance
289 const std::vector<std::pair<int64_t, int64_t>> &batchDimMap) { in verifyOutputShape() argument297 for (auto &dimPair : batchDimMap) in verifyOutputShape()407 auto batchDimMap = op.getBatchDimMap(); in verify() local418 if (!verifyDimMap(lhsType, rhsType, batchDimMap)) in verify()423 contractingDimMap, batchDimMap))) in verify()
1958 std::vector<std::pair<int64_t, int64_t>> batchDimMap = op.getBatchDimMap(); in matchAndRewrite() local1959 if (!batchDimMap.empty()) { in matchAndRewrite()1960 int64_t lhsIndex = batchDimMap[0].first; in matchAndRewrite()1961 int64_t rhsIndex = batchDimMap[0].second; in matchAndRewrite()