MCPcopy Create free account
hub / github.com/pytorch/pytorch / sync_fx_node

Method sync_fx_node

torch/_export/serde/serialize.py:1292–1297  ·  view source on GitHub ↗
(self, name: str, fx_node: torch.fx.Node)

Source from the content-addressed store, hash-verified

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)

Callers 5

deserialize_graphMethod · 0.95
deserialize_nodeMethod · 0.95
deserialize_outputsMethod · 0.95
generate_getitemMethod · 0.95

Calls 1

SerializeErrorClass · 0.85

Tested by

no test coverage detected