Searched defs:FlopCounterMode (Results 1 – 2 of 2) sorted by relevance
32 def FlopCounterMode(*args, **kwargs): function
590 class FlopCounterMode(TorchDispatchMode): class