Home
last modified time | relevance | path

Searched defs:conv_general_dilated (Results 1 – 3 of 3) sorted by relevance

/dports/math/py-jax/jax-0.2.9/jax/
H A Dlax_reference.py180 def conv_general_dilated(lhs, rhs, window_strides, padding, lhs_dilation, function
/dports/math/py-jax/jax-0.2.9/jax/experimental/jax2tf/tests/
H A Djax2tf_limitations.py315 def conv_general_dilated(cls, harness: primitive_harness.Harness): member in Jax2TfLimitation
/dports/math/py-jax/jax-0.2.9/jax/_src/lax/
H A Dlax.py516 def conv_general_dilated( function