def test_numpy_to_numpy_with_device(self): orig = self.orig_numpy() self.send_check_equal(orig, backend.CpuDevice())
def _generate_inputs(self): inputs = self.generate_inputs() _check_array_types(inputs, backend.CpuDevice(), 'generate_inputs') return inputs
def test_eq_numpy(self): assert backend.get_device(numpy) == backend.get_device(numpy) assert backend.CpuDevice() == backend.get_device(numpy)
def send_check_equal(self, orig, device_spec): device = backend.get_device(device_spec) converted = device.send(orig) numpy.testing.assert_array_equal(backend.CpuDevice().send(orig), backend.CpuDevice().send(converted)) return converted
def check_unrecognized(self, arg): device = backend.get_device_from_array(arg) assert device == backend.CpuDevice()
def test_module_numpy_device(self): self.check_device_spec_numpy(backend.CpuDevice())
def check_deserialize_by_passing_none(self, y, query): ret = self.deserializer(query, None) numpy.testing.assert_array_equal( backend.CpuDevice().send(ret), self.data)
def test_init(self): device = backend.CpuDevice() self.check_device(device)
def check_deserialize(self, y, query): ret = self.deserializer(query, y) numpy.testing.assert_array_equal( backend.CpuDevice().send(y), self.data) self.assertIs(ret, y)
def check_deserialize_none_value(self, y): ret = self.deserializer('y', None) numpy.testing.assert_array_equal(backend.CpuDevice().send(ret), self.data)
def setup_cpu(self): self._setup(backend.CpuDevice()) self.f.forward_cpu = mock.MagicMock(return_value=(self.y1, self.y2))
def _forward_expected(self, inputs): outputs = self.forward_expected(inputs) _check_array_types(outputs, backend.CpuDevice(), 'forward_expected') return outputs
def _generate_grad_grad_inputs(self, inputs_template): grad_grad_inputs = self.generate_grad_grad_inputs(inputs_template) _check_array_types( grad_grad_inputs, backend.CpuDevice(), 'generate_grad_grad_inputs') return grad_grad_inputs