(self, dev, is_train, use_graph, sequential)
| 424 | gpu_dev.ResetGraph() |
| 425 | |
| 426 | def _forward_helper(self, dev, is_train, use_graph, sequential): |
| 427 | self.generate_data(dev) |
| 428 | model = MLP(self.sgd) |
| 429 | model.compile([self.inputs], |
| 430 | is_train=is_train, |
| 431 | use_graph=use_graph, |
| 432 | sequential=sequential) |
| 433 | |
| 434 | self.get_params(model) |
| 435 | |
| 436 | out = model(self.inputs) |
| 437 | np_out = self.numpy_forward(self.data) |
| 438 | |
| 439 | np.testing.assert_array_almost_equal(tensor.to_numpy(out), np_out) |
| 440 | |
| 441 | def _train_one_batch_helper(self, dev, is_train, use_graph, sequential): |
| 442 | self.generate_data(dev) |
no test coverage detected