Searched defs:norm_first (Results 1 – 2 of 2) sorted by relevance
84 const bool norm_first, in transformer_encoder_layer_forward()
87 def __init__(self, dim_in=3, dim_out=2, down_scale=4, norm_first=False): argument