layer\_utils
============

.. List the submodules





.. Autodoc anything defined in the module itself

   TODO: WE DON'T USE THIS OPTION RIGHT NOW BUT WE CAN REACTIVATE IF WANTED
   We use :ignore-module-all: so sphinx does not document the same module twice, even if it is reimported
   For reimports that should be documented somewhere other than where they are defined, the re-imports
   __module__ should be manually overridden -- i.e. in the ``__init__.py`` which contains ``from xxx import YYY``,
   add in ``YYY.__module__ = __name__``.

.. automodule:: modelopt.torch.export.layer_utils
   :members:
   :undoc-members:

   .. Also show members without docstrings. Only members from __all__ are considered as per conf.py
   .. Ideally we should add docstrings for these members.


   .. Overview table of available classes in the module
   
   
   


   .. Overview table of available functions in the module
   
   
   .. rubric:: Functions

   .. autosummary::
      :nosignatures:
   
      build_attention_config
      build_conv_config
      build_decoder_config
      build_embedding_config
      build_fused_linear_config
      build_layernorm_config
      build_linear_config
      build_medusa_heads_config
      build_mlp_config
      build_moe_config
      build_qkv
      build_recurrent_config
      build_stacked_experts
      check_model_compatibility
      dup_kv_weight
      get_dtype
      get_enc_dec_models
      get_enc_dec_token_ids
      get_encoder_config
      get_experts_linear_names
      get_experts_list
      get_transformer_layers
      is_attention
      is_conv
      is_decoder_list
      is_embedding
      is_layernorm
      is_linear
      is_mlp
      is_moe
      is_quantlinear
      is_recurrent
      model_type_is_enc_dec
      update_experts_avg_prequant_scale