Searched refs:rhsBatchDimSet (Results 1 – 9 of 9) sorted by relevance
160 DenseSet<int64_t> rhsBatchDimSet; in verifyOutputShape() local162 rhsBatchDimSet.insert(dimPair.second); in verifyOutputShape()174 if (rhsContractingDimSet.count(i) > 0 || rhsBatchDimSet.count(i) > 0) in verifyOutputShape()
238 DenseSet<int64_t> rhsBatchDimSet; in verifyOutputShape() local240 rhsBatchDimSet.insert(dimPair.second); in verifyOutputShape()252 if (rhsContractingDimSet.count(i) > 0 || rhsBatchDimSet.count(i) > 0) in verifyOutputShape()
296 DenseSet<int64_t> rhsBatchDimSet; in verifyOutputShape() local298 rhsBatchDimSet.insert(dimPair.second); in verifyOutputShape()310 if (rhsContractingDimSet.count(i) > 0 || rhsBatchDimSet.count(i) > 0) in verifyOutputShape()
282 DenseSet<int64_t> rhsBatchDimSet; in verifyOutputShape() local284 rhsBatchDimSet.insert(dimPair.second); in verifyOutputShape()296 if (rhsContractingDimSet.count(i) > 0 || rhsBatchDimSet.count(i) > 0) in verifyOutputShape()
488 DenseSet<int64_t> rhsBatchDimSet; in verifyOutputShape() local490 rhsBatchDimSet.insert(dimPair.second); in verifyOutputShape()502 if (rhsContractingDimSet.count(i) > 0 || rhsBatchDimSet.count(i) > 0) in verifyOutputShape()
488 DenseSet<int64_t> rhsBatchDimSet;490 rhsBatchDimSet.insert(dimPair.second);502 if (rhsContractingDimSet.count(i) > 0 || rhsBatchDimSet.count(i) > 0)
512 DenseSet<int64_t> rhsBatchDimSet; in verifyOutputShape() local514 rhsBatchDimSet.insert(dimPair.second); in verifyOutputShape()526 if (rhsContractingDimSet.count(i) > 0 || rhsBatchDimSet.count(i) > 0) in verifyOutputShape()