Function gen_elemadd_data

imperative/python/test/unit/core/test_custom_op.py:79–84

(seed, shape, low=-1, high=1)

            return apply(op, ograd)

    def gen_elemadd_data(seed, shape, low=-1, high=1):
        rng = np.random.RandomState(seed=seed)
        lhs_np = rng.uniform(low=low, high=high, size=shape).astype(np.float32)
        rhs_np = rng.uniform(low=low, high=high, size=shape).astype(np.float32)
        ograd_np = rng.uniform(low=low, high=high, size=shape).astype(np.float32)
        return lhs_np, rhs_np, ograd_np

    def builtin_func(lhs, rhs, smooth):
        out = lhs + rhs