David Hou 1dbf3b2b19 Benchmarks for individual resnet layers (#4182) ...

* resnet individual layer benchmarks!

* small

* 1 and 2

* mem_used

* no ci

* better conv print

* defaults

* prints

* adjust

* adjust

* adjust

* benchmark only one layer example

* tensor.training, zero_grad, sum instead of mean, last mem, last kernel count

* default jitcnt=1

* scale flops/kernels with jitcnt

* add note about jitcnt memory

* touchup

2024-04-16 13:53:18 -04:00

2.1 KiB

Raw Blame History

View Raw