Method test_device_placement_cpu

tests/utils/test_device.py:83–85

(self)

        self.assertTrue(device.index == target_device_index)

    def test_device_placement_cpu(self):
        with device_placement(Frameworks.torch, 'cpu'):
            pass

    @unittest.skip('skip this test to avoid debug logging.')
    def test_device_placement_tf_gpu(self):