Function test_full_like

tests/python/relax/test_op_gradient_numeric.py:302–307

(target, dev)

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


(create_op_func,) = tvm.testing.parameters(