MCPcopy Create free account
hub / github.com/apache/tvm / simple_call

Method simple_call

tests/python/relax/test_relax_operators.py:456–458  ·  view source on GitHub ↗
(x: R.Tensor((3,), "float32"))

Source from the content-addressed store, hash-verified

454 class CallPyFuncTest:
455 @R.function
456 def simple_call(x: R.Tensor((3,), "float32")):
457 result = R.call_py_func(R.str("torch_relu"), (x,), out_sinfo=R.Tensor((3,), "float32"))
458 return result
459
460 @R.function
461 def multiple_calls(x: R.Tensor((2,), "float32")):

Callers

nothing calls this directly

Calls 2

TensorMethod · 0.80
call_py_funcMethod · 0.80

Tested by

no test coverage detected