RWKVBlock
=========

.. currentmodule:: tfts.models.rwkv


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

   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   .. rubric:: Methods

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

   
    
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   .. rubric:: Attributes

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