Exemple #1
0
def test_supports_device_gpu():
    assert NgraphBackend.supports_device('CUDA')
Exemple #2
0
def test_supports_device_gpu(backend_name):
    assert NgraphBackend.supports_device(backend_name)