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