test_topi_batch_matmul.py
1.73 KB
-
[Relay/TOPI][Op] Add batch_matmul in relay and TOPI (#2561) · 84590063
* Add batch_dot and cpu schedule * Add relay support for batch_dot * Rename batch_dot to batch_matmul * nits * Add missing file * Put batch_matmul and dense x86 schedule in separate files * Fix pylint * Remove unused import * Add cuda schedule for batch_matmul * Add test case with larger batch size * Add batch_matmul in api doc * Fix quantize pass rounding error * Fix pylint and minor change * bug fix
Haichen Shen committed