def forward(self, input):
yield stash("none", None)
return input
def forward(self, input):
yield stash("1to3", input)
output = self.conv(input)
return output
def forward(self, input):
yield stash("foo", input)
return input * 2
def forward(self, input):
yield stash("skip", input)
return input
def forward(self, input):
yield stash("bar", input)
return input
def forward(self, input):
foo = yield stash("foo")
return input + foo
def forward(self, x):
yield stash("hello", x)
return self.conv(x)
