Function _gen_weights

tests/python/relax/test_op_gradient_numeric.py:100–105

(out_sinfo)

        return data

    def _gen_weights(out_sinfo):
        if isinstance(out_sinfo, TupleStructInfo):
            return [_gen_weights(sinfo) for sinfo in out_sinfo.fields]
        else:
            assert isinstance(out_sinfo, TensorStructInfo)
            return np.random.uniform(size=[int(i) for i in out_sinfo.shape]).astype(out_sinfo.dtype)

    def _is_call_no_grad(expr):
        return isinstance(expr, Call) and expr.op == Op.get("relax.grad.no_grad")