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

Method main

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

Source from the content-addressed store, hash-verified

3156 class Expected_log10:
3157 @R.function
3158 def main(
3159 inp_0: R.Tensor((1, 3, 10, 10), dtype="float32"),
3160 ) -> R.Tensor((1, 3, 10, 10), dtype="float32"):
3161 with R.dataflow():
3162 lv: R.Tensor((1, 3, 10, 10), dtype="float32") = R.log(inp_0)
3163 lv1: R.Tensor((1, 3, 10, 10), dtype="float32") = R.divide(
3164 lv, R.const(2.302585092994046, "float32")
3165 )
3166 gv: R.Tensor((1, 3, 10, 10), dtype="float32") = lv1
3167 R.output(gv)
3168 return gv
3169
3170 verify_model(Log10(), input_info, {}, Expected_log10)
3171

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