Searched defs:batchDimMap (Results 1 – 2 of 2) sorted by relevance
289 const std::vector<std::pair<int64_t, int64_t>> &batchDimMap) { in verifyOutputShape()407 auto batchDimMap = op.getBatchDimMap(); in verify() local
1958 std::vector<std::pair<int64_t, int64_t>> batchDimMap = op.getBatchDimMap(); in matchAndRewrite() local