Home
last modified time | relevance | path

Searched refs:decoder_layer_options (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/options/
H A Dtransformer.cpp34 const TransformerDecoderLayerOptions& decoder_layer_options, in TransformerDecoderOptions() argument
36 : decoder_layer_(decoder_layer_options), num_layers_(num_layers) {} in TransformerDecoderOptions()
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/options/
H A Dtransformercoder.h62 const TransformerDecoderLayerOptions& decoder_layer_options,