Function test_mean

tests/python/relax/test_op_gradient_numeric.py:351–353

(target, dev)

@tvm.testing.parametrize_targets("llvm")
def test_mean(target, dev):
    data1_numpy = np.random.uniform(0, 16, (3, 3)).astype(np.float32)
    relax_check_gradients(relax.op.mean, [data1_numpy], target, dev)


@tvm.testing.parametrize_targets("llvm")