Method test_nonzero_gpu

test/python/test_onnx.py:1906–1907

(self)

    @unittest.skipIf(not singa_api.USE_CUDA, 'CUDA is not enabled')
    def test_nonzero_gpu(self):
        self._nonzero_helper(gpu_dev)

    def _cast_helper(self, dev):
        X = np.array([[1, 0], [1, 1]]).astype(np.float32)