(self, name: str, fx_node: torch.fx.Node)
| 1290 | ) |
| 1291 | |
| 1292 | def sync_fx_node(self, name: str, fx_node: torch.fx.Node): |
| 1293 | if name in self.serialized_name_to_node: |
| 1294 | raise SerializeError(f"Node {name} has already been deserialized before.") |
| 1295 | self.serialized_name_to_node[name] = fx_node |
| 1296 | assert "val" not in fx_node.meta |
| 1297 | fx_node.meta["val"] = self.serialized_name_to_meta[name] |
| 1298 | |
| 1299 | def deserialize_sym_op_inputs(self, inputs): |
| 1300 | return tuple(self.deserialize_input(input.arg) for input in inputs) |
no test coverage detected