(self, net, version='fp32', is_train=True)
| 517 | ) |
| 518 | |
| 519 | def _add_ops(self, net, version='fp32', is_train=True): |
| 520 | if self.evicted_values and is_train: |
| 521 | net.CopyRowsToTensor( |
| 522 | [self.w, self.evicted_values.get(), self.reinit_vec], [self.w]) |
| 523 | if _is_id_list(self.input_record): |
| 524 | self._add_ops_id_list(net, version=version) |
| 525 | elif _is_id_score_list(self.input_record): |
| 526 | self._add_ops_id_score_list(net, version=version) |
| 527 | else: |
| 528 | raise "Unsupported input type {0}".format(self.input_record) |
| 529 | |
| 530 | def add_train_ops(self, net): |
| 531 | self._add_ops(net, self.trainer_version, is_train=True) |
no test coverage detected