dmx.compressor.perf_proxy.PerformanceProxyMixin

class dmx.compressor.perf_proxy.PerformanceProxyMixin(*args, **kwargs)

__init__(*args, **kwargs) → None

Methods

`__init__`(args, *kwargs)
`count_flops`(_input, _output)
`counting_flops`([zero])
`enable_flop_counter`([state])
`zero_flop_counter`()

Attributes

`bops`
`flop_counter`
`flop_counter_enabled`
`flops`
`last_input_shape`
`last_output_shape`
`weight_elem_count`
`weight_size_in_bytes`