def bs_strided(x, n_filters):
     return nn.BackShiftConv3D(n_filters,
                               filter_size=filter_size,
                               **conv_args_strided)(x)
 def bs(x, n_filters):
     return nn.BackShiftConv3D(n_filters,
                               filter_size=filter_size,
                               **conv_args)(x)
 def bs(x):
     return nn.BackShiftConv3D(n_filters, use_bias=False, nln=tf.nn.elu)(x)