mirrors
/
mmengine
mirror of https://github.com/open-mmlab/mmengine.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
mmengine/tests/test_analysis
History
Mashiro 8beacd3b58
[Fix] Support calculate the flops of `matmul` with single dimension matrix (#970) 
* Support calculate the flops of matmul

* Remove unnecessary type ignore

* Update mmengine/analysis/jit_handles.py

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

---------

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
 		2023-03-09 17:29:26 +08:00
..
test_activation_count.py [Feature] Support model complexity computation (#779) 2023-02-20 15:00:28 +08:00
test_flop_count.py [Fix] Support calculate the flops of `matmul` with single dimension matrix (#970) 2023-03-09 17:29:26 +08:00
test_jit_analysis.py [Feature] Support model complexity computation (#779) 2023-02-20 15:00:28 +08:00
test_param_count.py [Feature] Support model complexity computation (#779) 2023-02-20 15:00:28 +08:00