Split parameter that has more than 1 dest ops
(self)
| 443 | self.graph.remove_variable(var) |
| 444 | |
| 445 | def format_parameter(self) -> None: |
| 446 | """ Split parameter that has more than 1 dest ops """ |
| 447 | for var in [_ for _ in self.graph.variables.values()]: |
| 448 | var.value = convert_any_to_torch_tensor(var.value) |
| 449 | if var.is_parameter and len(var.dest_ops) > 1: |
| 450 | for op in var.dest_ops: |
| 451 | created = self.graph.create_variable( |
| 452 | value=var.value.clone(), is_parameter=True) |
| 453 | op.inputs[op.inputs.index(var)] = created |
| 454 | created.dest_ops.append(op) |
| 455 | var.dest_ops.clear() |
| 456 | self.graph.remove_variable(var) |
| 457 | |
| 458 | def __delete_constant_input(self, op: Operation, input_idx: int): |
| 459 | op_name = op.name |
no test coverage detected