(f, *varargs, axis=None, edge_order=1)
| 77 | |
| 78 | |
| 79 | def gradient(f, *varargs, axis=None, edge_order=1): |
| 80 | xp = get_array_namespace(f) |
| 81 | return xp.gradient(f, *varargs, axis=axis, edge_order=edge_order) |
| 82 | |
| 83 | |
| 84 | def _dask_or_eager_func( |
searching dependent graphs…