mac_count.cc
5.04 KB
-
[RELAY][PASS] add a relay pass to count #macs of a model (#2609) · 615fd692
* add a relay pass to count #macs of a model * remove unnecessary includes * fix end-of-file issues * address review comments * remove test resnet * address more review comments * use data layout string to locate the input channel * fix bug in conv 2d output tensor checking * C must exist
Yida Wang committed