Function test_ones_zeros

tests/python/relax/test_op_gradient_numeric.py:317–320

(target, dev, create_op_func)

@tvm.testing.parametrize_targets("llvm")
def test_ones_zeros(target, dev, create_op_func):
    relax_check_gradients(
        create_op_func, [], target, dev, ignore_grads=[0], shape=(3, 3), dtype="float32"
    )


@tvm.testing.parametrize_targets("llvm")