projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 2626cd3) | patch
Max Pool with indices (#64121)
authorElias Ellison <eellison@devfair044.h1.fair>
Wed, 15 Sep 2021 20:43:12 +0000 (13:43 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Wed, 15 Sep 2021 20:45:46 +0000 (13:45 -0700)
commit29514bfcdbe460b15900b762576ed9bb1eea45d5
tree6c54894cd54e52c0ba96505d725bd455f75640c0tree | snapshot
parent2626cd3ba4452234ac82501d7959a4c5ca6bf063commit | diff
Max Pool with indices (#64121)

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

Add support for aten operators which return multiple outputs

Test Plan: Imported from OSS

Reviewed By: driazati

Differential Revision: D30738142

Pulled By: eellison

fbshipit-source-id: 0d7e51187bd5e3e9b43f0fdb5178366a97aec943
test/test_ops.py diff | blob | history
torch/csrc/jit/passes/symbolic_shape_analysis.cpp diff | blob | history
torch/csrc/jit/runtime/symbolic_shape_registry.cpp diff | blob | history
torch/testing/_internal/common_jit.py diff | blob | history
torch/testing/_internal/common_methods_invocations.py diff | blob | history
Domain: Machine Learning / ML Framework;