(self)
| 484 | operation.inputs.append(var) |
| 485 | |
| 486 | def remove_identity(self): |
| 487 | removing_ops = [] |
| 488 | for op in self.graph.operations.values(): |
| 489 | if op.type == 'Identity': removing_ops.append(op) |
| 490 | |
| 491 | for op in removing_ops: |
| 492 | self.graph.remove_operation(op, keep_coherence=True) |
| 493 | |
| 494 | def convert_to_tensor(self): |
| 495 | """ Convert anything inside your network to torch tensor. (Usually from numpy)""" |
no test coverage detected