Searched defs:batchDimMap (Results 1 – 2 of 2) sorted by relevance
974 std::vector<std::pair<int64_t, int64_t>> batchDimMap = op.getBatchDimMap(); in matchAndRewriteMaskableOp() local
868 verifyOutputShape(ContractionOp op,VectorType lhsType,VectorType rhsType,Type accType,Type resType,const std::vector<std::pair<int64_t,int64_t>> & contractingDimMap,const std::vector<std::pair<int64_t,int64_t>> & batchDimMap) verifyOutputShape() argument 995 auto batchDimMap = getBatchDimMap(); verify() local