Searched defs:get_total_flops (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/test/ | ||
| D | test_flop_counter.py | 36 def get_total_flops(mode): function |
| /external/pytorch/torch/utils/ | ||
| D | flop_counter.py | 630 def get_total_flops(self) -> int: member in FlopCounterMode |