Searched refs:custom_assert (Results 1 – 3 of 3) sorted by relevance
| /dports/math/py-jax/jax-0.2.9/jax/experimental/jax2tf/tests/ |
| H A D | jax2tf_limitations.py | 80 self.custom_assert = custom_assert
153 custom_assert=custom_assert,
296 custom_assert=custom_assert,
460 custom_assert=custom_assert,
562 custom_assert=custom_assert,
637 custom_assert=custom_assert,
702 custom_assert=custom_assert,
747 custom_assert=custom_assert,
772 custom_assert=custom_assert,
807 custom_assert=custom_assert,
[all …]
|
| H A D | tf_test_util.py | 186 custom_assert_lim = [l for l in jax2tf_limits if l.custom_assert]
191 custom_assert_lim[0].custom_assert(self, result_jax, result_tf, args=args, tol=max_tol)
|
| H A D | primitives_test.py | 182 elif l.skip_comparison or l.custom_assert:
|