Function compute_memory

tools/torchstat/compute_memory.py:6–22 · view source on GitHub ↗

(module, inp, out)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

_forward_post_hookFunction · 0.90

compute_ReLU_memoryFunction · 0.85

compute_PReLU_memoryFunction · 0.85

compute_Conv2d_memoryFunction · 0.85

compute_BatchNorm2d_memoryFunction · 0.85

compute_Linear_memoryFunction · 0.85

compute_Pool2d_memoryFunction · 0.85

no test coverage detected