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