Esempio n. 1
0
 def forward(self, input):
     yield stash("1to3", input)
     output = self.conv(input)
     return output  # noqa
Esempio n. 2
0
 def forward(self, input):
     yield stash("none", None)
     return input  # noqa
Esempio n. 3
0
 def forward(self, input):
     yield stash("skip", input)
     return input  # noqa
Esempio n. 4
0
 def forward(self, input):
     yield stash("bar", input)
     return input  # noqa: B901
Esempio n. 5
0
 def forward(self, input):
     foo = yield stash("foo")
     return input + foo
Esempio n. 6
0
 def forward(self, input):
     yield stash("foo", input)
     return input * 2  # noqa: B901
Esempio n. 7
0
 def forward(self, x):
     yield stash("hello", x)
     return self.conv(x)  # noqa