This is the complete list of members for Mila::Dnn::TransformerBlockConfig, including all inherited members.

activation_type_	Mila::Dnn::TransformerBlockConfig	private
dropout_	Mila::Dnn::TransformerBlockConfig	private
getActivationType() const	Mila::Dnn::TransformerBlockConfig	inline
getDropout() const	Mila::Dnn::TransformerBlockConfig	inline
getHiddenDimension() const	Mila::Dnn::TransformerBlockConfig	inline
getInputShape() const	Mila::Dnn::TransformerBlockConfig	inline
getName() const	Mila::Dnn::ComponentConfig	inline
getNumHeads() const	Mila::Dnn::TransformerBlockConfig	inline
getPrecision() const	Mila::Dnn::ComponentConfig	inline
hidden_dim_	Mila::Dnn::TransformerBlockConfig	private
input_shape_	Mila::Dnn::TransformerBlockConfig	private
is_training_	Mila::Dnn::ComponentConfig	protected
isTraining() const	Mila::Dnn::ComponentConfig	inline
name_	Mila::Dnn::ComponentConfig	protected
num_heads_	Mila::Dnn::TransformerBlockConfig	private
precision_	Mila::Dnn::ComponentConfig	protected
TransformerBlockConfig(const std::vector< size_t > &input_shape, size_t num_heads)	Mila::Dnn::TransformerBlockConfig	inline
use_bias_	Mila::Dnn::TransformerBlockConfig	private
use_pre_ln_	Mila::Dnn::TransformerBlockConfig	private
useBias() const	Mila::Dnn::TransformerBlockConfig	inline
usePreLayerNorm() const	Mila::Dnn::TransformerBlockConfig	inline
validate() const	Mila::Dnn::TransformerBlockConfig	inlinevirtual
withActivation(ActivationType activation_type)	Mila::Dnn::TransformerBlockConfig	inline
withBias(bool use_bias)	Mila::Dnn::TransformerBlockConfig	inline
withDropout(float dropout)	Mila::Dnn::TransformerBlockConfig	inline
withHiddenDimension(size_t hidden_dim)	Mila::Dnn::TransformerBlockConfig	inline
withName(this Self &&self, std::string name)	Mila::Dnn::ComponentConfig	inline
withPrecision(this Self &&self, ComputePrecision::Policy policy)	Mila::Dnn::ComponentConfig	inline
withPreLayerNorm(bool use_pre_ln)	Mila::Dnn::TransformerBlockConfig	inline
withTraining(this Self &&self, bool is_training)	Mila::Dnn::ComponentConfig	inline
~ComponentConfig()=default	Mila::Dnn::ComponentConfig	virtual