pytorch/TestJit.test_import_method.expect at main - pytorch - Gitea: Git for Me

frozenleaves/pytorch

mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00

Files

Tugsbayasgalan (Tugsuu) Manlaibaatar b0c27b44cf Enable backward/forward compatibility for TS runtime (#57498 )

Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/57498

Test Plan: Imported from OSS

Reviewed By: albanD

Differential Revision: D28162448

Pulled By: tugsbayasgalan

fbshipit-source-id: 5c21ced42a22aca7cee089e876e9d98d32f68955

2021-05-07 15:41:45 -07:00

5 lines

98 B

Plaintext

Raw Permalink Blame History

 def forward(self,
     x: Tensor,
     y: Tensor) -> Tensor:
   return torch.add(torch.mul(x, 2), y)