dmx.compressor.modeling.nn.core.DmxModule ========================================= .. currentmodule:: dmx.compressor.modeling.nn.core .. autoclass:: DmxModule .. automethod:: __init__ .. rubric:: Methods .. autosummary:: ~DmxModule.__init__ ~DmxModule.add_module ~DmxModule.align_device ~DmxModule.apply ~DmxModule.approx_forward ~DmxModule.approximator_wrapper ~DmxModule.bfloat16 ~DmxModule.buffers ~DmxModule.calibrating_quantizers ~DmxModule.calibrating_smoothquant ~DmxModule.check_format_dim_consistency ~DmxModule.check_input_format_dim_consistency ~DmxModule.check_residual_format_dim_consistency ~DmxModule.check_sparseness_dim_consistency ~DmxModule.check_weight_format_dim_consistency ~DmxModule.children ~DmxModule.compile ~DmxModule.configure ~DmxModule.count_flops ~DmxModule.counting_flops ~DmxModule.cpu ~DmxModule.cuda ~DmxModule.dmx_config ~DmxModule.double ~DmxModule.enable_approximation_function_tuning ~DmxModule.enable_flop_counter ~DmxModule.enable_optimal_brain_compression ~DmxModule.enable_quantizer_calib ~DmxModule.enable_smoothquant_calib ~DmxModule.eval ~DmxModule.extra_repr ~DmxModule.float ~DmxModule.fold_weight_and_bias ~DmxModule.forward ~DmxModule.get_buffer ~DmxModule.get_extra_state ~DmxModule.get_parameter ~DmxModule.get_submodule ~DmxModule.half ~DmxModule.infer_ch_axis ~DmxModule.init_casts ~DmxModule.init_smoothquant ~DmxModule.init_sparsifier ~DmxModule.ipu ~DmxModule.load_state_dict ~DmxModule.load_state_dict_and_register_url ~DmxModule.measuring_runtime ~DmxModule.modules ~DmxModule.monitoring ~DmxModule.mtia ~DmxModule.named_buffers ~DmxModule.named_children ~DmxModule.named_modules ~DmxModule.named_parameters ~DmxModule.optimal_brain_compressing ~DmxModule.parameters ~DmxModule.register_backward_hook ~DmxModule.register_buffer ~DmxModule.register_forward_hook ~DmxModule.register_forward_pre_hook ~DmxModule.register_full_backward_hook ~DmxModule.register_full_backward_pre_hook ~DmxModule.register_load_state_dict_post_hook ~DmxModule.register_load_state_dict_pre_hook ~DmxModule.register_module ~DmxModule.register_parameter ~DmxModule.register_state_dict_post_hook ~DmxModule.register_state_dict_pre_hook ~DmxModule.requires_grad_ ~DmxModule.save_state_dict_and_register_url ~DmxModule.set_extra_state ~DmxModule.set_submodule ~DmxModule.share_memory ~DmxModule.slanc_tuning ~DmxModule.state_dict ~DmxModule.to ~DmxModule.to_compiler_graph ~DmxModule.to_empty ~DmxModule.train ~DmxModule.transform ~DmxModule.tuning_approximation_function ~DmxModule.type ~DmxModule.update_params_with_raw ~DmxModule.update_smoothquant_scale ~DmxModule.xpu ~DmxModule.zero_flop_counter ~DmxModule.zero_grad .. rubric:: Attributes .. autosummary:: ~DmxModule.T_destination ~DmxModule.accum_format ~DmxModule.approximation_function ~DmxModule.bias_format ~DmxModule.bops ~DmxModule.call_super_init ~DmxModule.dump_patches ~DmxModule.effective_weight ~DmxModule.flop_counter ~DmxModule.flop_counter_enabled ~DmxModule.flops ~DmxModule.functional_forward ~DmxModule.input_formats ~DmxModule.input_precision ~DmxModule.is_compound ~DmxModule.last_input_shape ~DmxModule.last_output_shape ~DmxModule.multiplier_format ~DmxModule.output_formats ~DmxModule.plugins ~DmxModule.residual_format ~DmxModule.weight_elem_count ~DmxModule.weight_format ~DmxModule.weight_hypernet ~DmxModule.weight_precision ~DmxModule.weight_scale ~DmxModule.weight_size_in_bytes ~DmxModule.weight_sparseness ~DmxModule.weight_storage_format ~DmxModule.weight_storage_precision ~DmxModule.weight_storage_scale ~DmxModule.weight_storage_zero_point ~DmxModule.weight_zero_point ~DmxModule.training