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

Method main

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

Source from the content-addressed store, hash-verified

3134 class Expected_log2:
3135 @R.function
3136 def main(
3137 inp_0: R.Tensor((1, 3, 10, 10), dtype="float32"),
3138 ) -> R.Tensor((1, 3, 10, 10), dtype="float32"):
3139 with R.dataflow():
3140 lv: R.Tensor((1, 3, 10, 10), dtype="float32") = R.log(inp_0)
3141 lv1: R.Tensor((1, 3, 10, 10), dtype="float32") = R.divide(
3142 lv, R.const(0.6931471805599453, "float32")
3143 )
3144 gv: R.Tensor((1, 3, 10, 10), dtype="float32") = lv1
3145 R.output(gv)
3146 return gv
3147
3148 verify_model(Log2(), input_info, {}, Expected_log2)
3149

Callers

nothing calls this directly

Calls 5

TensorMethod · 0.80
dataflowMethod · 0.80
outputMethod · 0.80
logMethod · 0.45
divideMethod · 0.45

Tested by

no test coverage detected