(self)
| 55 | return net.SerializeToString() |
| 56 | |
| 57 | def test_run(self): |
| 58 | A = np.ones((2, 3), np.float32) |
| 59 | B = np.ones((3, 4), np.float32) |
| 60 | predictor = workspace.Predictor(self.init_net, self.predict_net) |
| 61 | outputs = predictor.run([A, B]) |
| 62 | self.assertEqual(len(outputs), 1) |
| 63 | np.testing.assert_almost_equal(np.dot(A, B), outputs[0]) |
| 64 | |
| 65 | def test_run_map(self): |
| 66 | A = np.zeros((2, 3), np.float32) |
nothing calls this directly
no test coverage detected