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

Method glob_func_2

tests/python/relax/test_transform_lambda_lift.py:309–320  ·  view source on GitHub ↗
(
            x1: R.Tensor((10, 5), "float32"), y1: R.Tensor((10, 5), "float32")
        )

Source from the content-addressed store, hash-verified

307
308 @R.function
309 def glob_func_2(
310 x1: R.Tensor((10, 5), "float32"), y1: R.Tensor((10, 5), "float32")
311 ) -> R.Tensor((10, 5), "float32"):
312 @R.function
313 def inner(
314 x2: R.Tensor((10, 5), "float32"), y2: R.Tensor((10, 5), "float32")
315 ) -> R.Tensor((10, 5), "float32"):
316 s: R.Tensor((10, 5), "float32") = R.add(x2, y2)
317 return s
318
319 gv1: R.Tensor((10, 5), "float32") = inner(x1, y1)
320 return gv1
321
322 before = Before
323 expected = Expected

Callers

nothing calls this directly

Calls 2

innerFunction · 0.85
TensorMethod · 0.80

Tested by

no test coverage detected