Function test_triu

tests/python/relax/test_op_gradient_numeric.py:324–326

(target, dev)

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


##################### Statistical #####################