Function test_ones_zeros_like

tests/python/relax/test_op_gradient_numeric.py:296–298

(target, dev, like_op_func)

@tvm.testing.parametrize_targets("llvm")
def test_ones_zeros_like(target, dev, like_op_func):
    data_numpy = np.random.uniform(-1, 1, (3, 3)).astype(np.float32)
    relax_check_gradients(like_op_func, [data_numpy], target, dev, ignore_grads=[0])


@tvm.testing.parametrize_targets("llvm")