Simple addition function.
(self, x, y)
| 32 | |
| 33 | @I.pyfunc |
| 34 | def add(self, x, y): |
| 35 | """Simple addition function.""" |
| 36 | x_tvm = self._convert_pytorch_to_tvm(x) |
| 37 | y_tvm = self._convert_pytorch_to_tvm(y) |
| 38 | result = self.call_tir(self.add_tir, [x_tvm, y_tvm], out_sinfo=R.Tensor((5,), "float32")) |
| 39 | return self._convert_tvm_to_pytorch(result) |
| 40 | |
| 41 | @I.pyfunc |
| 42 | def multiply(self, x, y): |
no test coverage detected