Searched defs:requires_gpu (Results 1 – 1 of 1) sorted by relevance
94 def test_conv1d(self, kwargs, expected_output_shape, requires_gpu=False): argument252 requires_gpu=False, argument366 def test_conv3d(self, kwargs, expected_output_shape=None, requires_gpu=False): argument