(self, dev)
| 1881 | self._tile_helper(gpu_dev) |
| 1882 | |
| 1883 | def _nonzero_helper(self, dev): |
| 1884 | X = np.array([[1, 0], [1, 1]]).astype(np.float32) |
| 1885 | x = tensor.from_numpy(X) |
| 1886 | x.to_device(dev) |
| 1887 | y = autograd.nonzero(x) |
| 1888 | |
| 1889 | # frontend |
| 1890 | model = sonnx.to_onnx([x], [y]) |
| 1891 | # print('The model is:\n{}'.format(model)) |
| 1892 | |
| 1893 | # backend |
| 1894 | sg_ir = sonnx.prepare(model, device=dev) |
| 1895 | sg_ir.is_graph = True |
| 1896 | y_t = sg_ir.run([x]) |
| 1897 | |
| 1898 | np.testing.assert_array_almost_equal( |
| 1899 | tensor.to_numpy(y).shape, |
| 1900 | tensor.to_numpy(y_t[0]).shape) |
| 1901 | |
| 1902 | def test_nonzero_cpu(self): |
| 1903 | self._nonzero_helper(cpu_dev) |
no test coverage detected