mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
517aee5369
Test Plan:
test script:
```
def test_zhxchen17(self):
from libfb.py.pyinit import initFacebook
initFacebook()
class M(torch.nn.Module):
def forward(self, x):
return torch.add(x, x)
def tmptmp(x, y):
return torch.mul(x, y)
m = M()
n = torch.jit.script(m)
print(n(torch.tensor(1)))
print(torch.jit.script(tmptmp)(torch.tensor(1), torch.tensor(2)))
```
```
I0802 12:01:23.932929 4079081 init.cc:407] Logging to scuba: run __torch__.caffe2.test.export.test_export.M.forward sample rate: 1000000
```
Differential Revision: D60920867
Pull Request resolved: https://github.com/pytorch/pytorch/pull/133484
Approved by: https://github.com/davidberard98
|
||
|---|---|---|
| .. | ||
| compilation_unit.h | ||
| function_impl.cpp | ||
| function_impl.h | ||
| method.h | ||
| module_save.cpp | ||
| module.cpp | ||
| module.h | ||
| object.cpp | ||
| object.h | ||