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