Ejemplo n.º 1
0
def test_supports_device_gpu():
    assert NgraphBackend.supports_device('CUDA')
Ejemplo n.º 2
0
def test_supports_device_gpu(backend_name):
    assert NgraphBackend.supports_device(backend_name)