コード例 #1
0
 def forward(self, input):
     yield stash("1to3", input)
     output = self.conv(input)
     return output  # noqa
コード例 #2
0
 def forward(self, input):
     yield stash("none", None)
     return input  # noqa
コード例 #3
0
ファイル: test_leak.py プロジェクト: yuhc/ava-pytorch
 def forward(self, input):
     yield stash("skip", input)
     return input  # noqa
コード例 #4
0
 def forward(self, input):
     yield stash("bar", input)
     return input  # noqa: B901
コード例 #5
0
 def forward(self, input):
     foo = yield stash("foo")
     return input + foo
コード例 #6
0
 def forward(self, input):
     yield stash("foo", input)
     return input * 2  # noqa: B901
コード例 #7
0
ファイル: test_api.py プロジェクト: yuhc/ava-pytorch
 def forward(self, x):
     yield stash("hello", x)
     return self.conv(x)  # noqa