Searched refs:TransformerDecoderLayerImpl (Results 1 – 2 of 2) sorted by relevance
93 TransformerDecoderLayerImpl::TransformerDecoderLayerImpl( in TransformerDecoderLayerImpl() function in torch::nn::TransformerDecoderLayerImpl100 void TransformerDecoderLayerImpl::reset() { in reset()146 void TransformerDecoderLayerImpl::reset_parameters() { in reset_parameters()164 Tensor TransformerDecoderLayerImpl::forward( in forward()200 Tensor TransformerDecoderLayerImpl::activation(const Tensor& input) { in activation()314 layers->at<TransformerDecoderLayerImpl>(i).reset_parameters(); in reset_parameters()339 output = layers->at<TransformerDecoderLayerImpl>(0).forward( in forward()348 output = layers->at<TransformerDecoderLayerImpl>(i).forward( in forward()
111 class TORCH_API TransformerDecoderLayerImpl112 : public Cloneable<TransformerDecoderLayerImpl> {114 TransformerDecoderLayerImpl(int64_t d_model, int64_t nhead) in TransformerDecoderLayerImpl() function115 : TransformerDecoderLayerImpl( in TransformerDecoderLayerImpl()117 explicit TransformerDecoderLayerImpl(TransformerDecoderLayerOptions options_);