[Static Runtime] Implement aten::nonzero out variant (#64126)
authorHarut Movsisyan <harutm@fb.com>
Tue, 31 Aug 2021 07:49:39 +0000 (00:49 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Tue, 31 Aug 2021 07:51:15 +0000 (00:51 -0700)
commit3c15822f5f4ab616eb6a519a0ff9b82fc7a3dc63
tree037d9c16423152ffce8157612cf2914057f8c8dd
parenta3d6dae319f03e9b5450c875349dbd65cb437767
[Static Runtime] Implement aten::nonzero out variant (#64126)

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

Test Plan:
Confirm out variant is called:

```
> buck run //caffe2/benchmarks/static_runtime:static_runtime_cpptest -- --v=1
```

Reviewed By: mikeiovine

Differential Revision: D30617729

fbshipit-source-id: 752749638c8f467815efa57021cb3de5c728ab1b
benchmarks/static_runtime/test_scripts.h
benchmarks/static_runtime/test_static_runtime.cc
torch/csrc/jit/runtime/static/ops.cpp