MoELayer
========

.. currentmodule:: tfts.layers.moe_layer


.. autoclass:: MoELayer
   :members:
   :show-inheritance:
   :exclude-members: __init__
   
   
   :inherited-members:
   

   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   .. rubric:: Methods

   .. autosummary::
   
      ~MoELayer.add_loss
      ~MoELayer.add_metric
      ~MoELayer.add_variable
      ~MoELayer.add_weight
      ~MoELayer.build
      ~MoELayer.build_from_config
      ~MoELayer.call
      ~MoELayer.compute_mask
      ~MoELayer.compute_output_shape
      ~MoELayer.compute_output_spec
      ~MoELayer.count_params
      ~MoELayer.from_config
      ~MoELayer.get_build_config
      ~MoELayer.get_config
      ~MoELayer.get_weights
      ~MoELayer.load_own_variables
      ~MoELayer.quantize
      ~MoELayer.quantized_build
      ~MoELayer.quantized_call
      ~MoELayer.rematerialized_call
      ~MoELayer.save_own_variables
      ~MoELayer.set_weights
      ~MoELayer.stateless_call
      ~MoELayer.symbolic_call
   
   

   
    
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~MoELayer.compute_dtype
      ~MoELayer.dtype
      ~MoELayer.dtype_policy
      ~MoELayer.input
      ~MoELayer.input_dtype
      ~MoELayer.input_spec
      ~MoELayer.losses
      ~MoELayer.metrics
      ~MoELayer.metrics_variables
      ~MoELayer.non_trainable_variables
      ~MoELayer.non_trainable_weights
      ~MoELayer.output
      ~MoELayer.path
      ~MoELayer.quantization_mode
      ~MoELayer.supports_masking
      ~MoELayer.trainable
      ~MoELayer.trainable_variables
      ~MoELayer.trainable_weights
      ~MoELayer.variable_dtype
      ~MoELayer.variables
      ~MoELayer.weights