You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Currently DynamicConvOp supports computed padding. We will also need computed window_strides, lhs_dilation and rhs_dilation.
I also notice that the op takes padding both as an input and as an attribute. JAX does not use the padding attribute, but I thought I'd point this out in case it is a mistake.
The text was updated successfully, but these errors were encountered:
I noticed that JAX's batching rules will end up making batch_group_count and feature_group_count dynamic if some of the input dimensions are dynamic. Hence, we need dynamic support for these also.
Request description
Currently DynamicConvOp supports computed padding. We will also need computed
window_strides
,lhs_dilation
andrhs_dilation
.I also notice that the op takes
padding
both as an input and as an attribute. JAX does not use thepadding
attribute, but I thought I'd point this out in case it is a mistake.The text was updated successfully, but these errors were encountered: