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

Method main

tests/python/relax/test_frontend_from_exported_program.py:605–614  ·  view source on GitHub ↗
(
            inp_0: R.Tensor((1, 3, 10, 10), dtype="float32"),
        )

Source from the content-addressed store, hash-verified

603 class Expected_log1p:
604 @R.function
605 def main(
606 inp_0: R.Tensor((1, 3, 10, 10), dtype="float32"),
607 ) -> R.Tuple(R.Tensor((1, 3, 10, 10), dtype="float32")):
608 with R.dataflow():
609 lv: R.Tensor((1, 3, 10, 10), dtype="float32") = R.log(
610 R.add(inp_0, R.const(1, "float32"))
611 )
612 gv: R.Tuple(R.Tensor((1, 3, 10, 10), dtype="float32")) = (lv,)
613 R.output(gv)
614 return gv
615
616 verify_model(Log1p(), example_args, {}, Expected_log1p)
617

Callers

nothing calls this directly

Calls 5

TensorMethod · 0.80
dataflowMethod · 0.80
outputMethod · 0.80
logMethod · 0.45
addMethod · 0.45

Tested by

no test coverage detected