Searched defs:FlopCounterMode (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/test/ | ||
| D | test_flop_counter.py | 32 def FlopCounterMode(*args, **kwargs): function |
| /external/pytorch/torch/utils/ | ||
| D | flop_counter.py | 590 class FlopCounterMode(TorchDispatchMode): class |