↓ 4 callersMethod__init__(self, in_channels, left_att, blocks, expanse_ratio, backbone_channels)
core/aggregation_fasternet.py:7
↓ 2 callersFunction_sample_1dSample a [N,C,1,W] tensor with pixel x coordinates shaped [N,1,K,1].
core/liteanystereov2_H.py:33
Method__init__(self, corr_levels, corr_radius, volume_dim, hidden_dim, motion_corr_dim, motion_disp_dim)
core/liteanystereov2_H.py:218
Method__init__(self, inp, oup, stride, mlp_ratio, n_div=4, act_layer=nn.GELU)
core/aggregation_fasternet.py:88
Method__init__(self, dim, mlp_ratio, n_div=4, act_layer=nn.GELU, layer_scale_init_value=0.0)
core/aggregation_fasternet.py:108