Searched refs:TransformerEncoderLayerImpl (Results 1 – 2 of 2) sorted by relevance
36 class TORCH_API TransformerEncoderLayerImpl37 : public Cloneable<TransformerEncoderLayerImpl> {39 TransformerEncoderLayerImpl(int64_t d_model, int64_t nhead) in TransformerEncoderLayerImpl() function40 : TransformerEncoderLayerImpl( in TransformerEncoderLayerImpl()42 explicit TransformerEncoderLayerImpl(TransformerEncoderLayerOptions options_);
15 TransformerEncoderLayerImpl::TransformerEncoderLayerImpl( in TransformerEncoderLayerImpl() function in torch::nn::TransformerEncoderLayerImpl22 void TransformerEncoderLayerImpl::reset() { in reset()48 void TransformerEncoderLayerImpl::reset_parameters() { in reset_parameters()63 Tensor TransformerEncoderLayerImpl::forward( in forward()247 layers->at<TransformerEncoderLayerImpl>(i).reset_parameters(); in reset_parameters()269 output = layers->at<TransformerEncoderLayerImpl>(0).forward( in forward()273 output = layers->at<TransformerEncoderLayerImpl>(i).forward( in forward()