MCPcopy Create free account
hub / github.com/horseee/DeepCache / rnn_flops_counter_hook

Function rnn_flops_counter_hook

DeepCache/flops.py:155–185  ·  view source on GitHub ↗

Takes into account batch goes at first position, contrary to pytorch common rule (but actually it doesn't matter). If sigmoid and tanh are hard, only a comparison FLOPS should be accurate

(rnn_module, input, output)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers

nothing calls this directly

Calls 1

rnn_flopsFunction · 0.85

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…