MCPcopy Create free account
hub / github.com/pytorch/examples / forward

Method forward

fx/custom_tracer.py:95–96  ·  view source on GitHub ↗
(self, a, b)

Source from the content-addressed store, hash-verified

93
94class M2(torch.nn.Module):
95 def forward(self, a, b):
96 return a + b
97
98class TaggingTracer(Tracer):
99 def create_node(self, kind : str, target : Union[str, Callable],

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected