Function test_sum_with_axis

tests/python/relax/test_op_gradient_numeric.py:339–341

(target, dev)

@tvm.testing.parametrize_targets("llvm")
def test_sum_with_axis(target, dev):
    data1_numpy = np.random.uniform(0, 16, (2, 3, 4, 5)).astype(np.float32)
    relax_check_gradients(relax.op.sum, [data1_numpy], target, dev, axis=[1, 3])


@tvm.testing.parametrize_targets("llvm")