Searched defs:num_encoder_layers (Results 1 – 3 of 3) sorted by relevance
87 def __init__(self, inp_vocab_size=10, model_dim=32, num_encoder_layers=2): argument
44 int64_t num_encoder_layers, in TransformerOptions()
502 num_encoder_layers=6, argument