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

Method add

tests/python/relax/test_tvmscript_parser.py:1836–1841  ·  view source on GitHub ↗
(
            X: T.Buffer([T.int64(2), T.int64(4)], "float32"),
            Y: T.Buffer((), "float32"),
            Z: T.Buffer([T.int64(2), T.int64(4)], "float32"),
        )

Source from the content-addressed store, hash-verified

1834 class Module:
1835 @T.prim_func(s_tir=True)
1836 def add(
1837 X: T.Buffer([T.int64(2), T.int64(4)], "float32"),
1838 Y: T.Buffer((), "float32"),
1839 Z: T.Buffer([T.int64(2), T.int64(4)], "float32"),
1840 ):
1841 T.evaluate(0)
1842
1843 @R.function
1844 def main(x: R.Tensor((2, 4), dtype="float32")) -> R.Tensor((10,), dtype="float32"):

Callers 15

fFunction · 0.45
fooFunction · 0.45
test_relax_tensor_opFunction · 0.45
test_shadowingFunction · 0.45
test_tuple_get_itemFunction · 0.45
fooMethod · 0.45
test_tensor_with_vdeviceFunction · 0.45
inner_funcFunction · 0.45
fooMethod · 0.45
fooMethod · 0.45
funcFunction · 0.45

Calls 1

evaluateMethod · 0.45

Tested by

no test coverage detected