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

Method main

tests/python/relax/test_transform_legalize_ops_binary.py:1034–1036  ·  view source on GitHub ↗
(x: R.Tensor((1, 2, 3), "float32"), y: R.Tensor((4, 3, 2, 1), "float32"))

Source from the content-addressed store, hash-verified

1032 class Equal:
1033 @R.function
1034 def main(x: R.Tensor((1, 2, 3), "float32"), y: R.Tensor((4, 3, 2, 1), "float32")) -> R.Tensor((4, 3, 2, 3), "bool"):
1035 gv: R.Tensor((4, 3, 2, 3), "bool") = R.equal(x, y)
1036 return gv
1037
1038 @tvm.script.ir_module
1039 class Expected:

Callers

nothing calls this directly

Calls 2

TensorMethod · 0.80
equalMethod · 0.45

Tested by

no test coverage detected