Function test_binary_arith

tests/python/relax/test_op_gradient_numeric.py:243–246

(target, dev, binary_arith_op_func)

@tvm.testing.parametrize_targets("llvm")
def test_binary_arith(target, dev, binary_arith_op_func):
    data1_numpy = np.random.uniform(1, 2, (3, 3)).astype(np.float32)
    data2_numpy = np.random.uniform(1, 2, (3, 3)).astype(np.float32)
    relax_check_gradients(binary_arith_op_func, [data1_numpy, data2_numpy], target, dev)


(binary_minmax_op_func,) = tvm.testing.parameters(