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

Class MyElementwiseModule

fx/native_interpreter/use_interpreter.py:114–116  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

112 return torch.fx.GraphModule(wrapper, graph)
113
114class MyElementwiseModule(torch.nn.Module):
115 def forward(self, x, y):
116 return x * y + y
117
118mem = MyElementwiseModule()
119lowered = lower_to_elementwise_interpreter(mem)

Callers 1

use_interpreter.pyFile · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected