MCPcopy Index your code
hub / github.com/apache/tvm / outer_func

Function outer_func

tests/python/relax/test_tvmscript_parser.py:1095–1103  ·  view source on GitHub ↗
(
            c1: R.Tensor((2, 3), "float32"),
        )

Source from the content-addressed store, hash-verified

1093 ):
1094 @R.function
1095 def outer_func(
1096 c1: R.Tensor((2, 3), "float32"),
1097 ) -> R.Callable((R.Tensor(None, "float32", ndim=2),), R.Tensor(None, "float32", ndim=2)):
1098 @R.function
1099 def inner_func(x1: R.Tensor((2, 3), "float32")):
1100 s: R.Tensor((2, 3), "float32") = R.add(x1, c1)
1101 return s
1102
1103 return inner_func
1104
1105 in_call = outer_func(x)
1106 res = in_call(y)

Callers 2

mainMethod · 0.85
mainFunction · 0.85

Calls 1

TensorMethod · 0.80

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…