projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 227cb26) | patch
[Static Runtime] Implement prim::VarStack out variant (#63579)
authorMike Iovine <mikeiovine@fb.com>
Tue, 24 Aug 2021 16:38:25 +0000 (09:38 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Tue, 24 Aug 2021 16:44:29 +0000 (09:44 -0700)
commit7774a4e95b69d7a61ae1644a27f9b924d93ca600
tree56a19a1b01c580054884f460500467bae4e678f3tree | snapshot
parent227cb268bccd22feb8aa8651773a202ec1e09c7fcommit | diff
[Static Runtime] Implement prim::VarStack out variant (#63579)

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

Provide a static runtime out variant implementation for the new op introduced in D30426232 (https://github.com/pytorch/pytorch/commit/1385f9fb12e6607c98d2d9d5edaaaab2bc07386f).

Test Plan: `buck test //caffe2/benchmarks/static_runtime:static_runtime_cpptest -- IndividualOps_VarStack`

Reviewed By: navahgar

Differential Revision: D30410525

fbshipit-source-id: bc59a3d8ad23e3d94561ec2dca9cc20687dbadf8
benchmarks/static_runtime/test_scripts.h diff | blob | history
benchmarks/static_runtime/test_static_runtime.cc diff | blob | history
torch/csrc/jit/runtime/static/impl.cpp diff | blob | history
torch/csrc/jit/runtime/static/ops.cpp diff | blob | history
Domain: Machine Learning / ML Framework;