dmx.compressor.perf_proxy

Classes

PerformanceProxyMixin(*args, **kwargs)

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

Bases: object

property bops: float | None

count_flops(_input: Tensor, _output: Tensor) → None

counting_flops(zero: bool = True) → None

enable_flop_counter(state: bool = True) → None

flop_counter: int | None = None

flop_counter_enabled: bool = False

property flops: float | None

last_input_shape: Size | None = None

last_output_shape: Size | None = None

property weight_elem_count: float | None

property weight_size_in_bytes: float | None

zero_flop_counter() → None