コード例 #1
0
ファイル: test_gpipe.py プロジェクト: hulaba/fairscale
 def forward(self, input):
     skip_1to3 = yield pop("1to3")
     output = self.conv(input) + skip_1to3
     return output
コード例 #2
0
ファイル: test_gpipe.py プロジェクト: hulaba/fairscale
 def forward(self, input):
     none = yield pop("none")
     assert none is None
     return input
コード例 #3
0
ファイル: test_stash_pop.py プロジェクト: zzszmyf/fairscale
 def forward(self, input):
     foo = yield pop("foo")
     return foo
コード例 #4
0
ファイル: test_stash_pop.py プロジェクト: zzszmyf/fairscale
 def forward(self, input):
     yield pop("foo")
コード例 #5
0
 def forward(self, input):
     skip = yield pop("skip")
     return input + skip
コード例 #6
0
 def forward(self, input):
     bar = yield pop("bar")
     return input + bar