Searched refs:getBatchDimMap (Results 1 – 3 of 3) sorted by relevance
760 auto batchDimMap = getBatchDimMap(); in verify()877 std::vector<std::pair<int64_t, int64_t>> ContractionOp::getBatchDimMap() { in getBatchDimMap() function in ContractionOp
1803 std::vector<std::pair<int64_t, int64_t>> batchDimMap = op.getBatchDimMap(); in matchAndRewrite()
251 std::vector<std::pair<int64_t, int64_t>> getBatchDimMap();