Allow dispatch based on tensor list args (#17522)
authorSebastian Messmer <messmer@fb.com>
Fri, 1 Mar 2019 00:25:37 +0000 (16:25 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Fri, 1 Mar 2019 00:32:00 +0000 (16:32 -0800)
commit2e94054e347d25d74506c9095b56b83d923f2a60
treeef279cb1fc2c1beb59e0c46b111b9ae71d261dd2
parentb004b31d06c0fcfaf96647763fc5a13f72105494
Allow dispatch based on tensor list args (#17522)

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

Dispatch is still based on the first tensor arg, but that first "tensor arg" is now allowed to be a tensor list.
That is, the first argument that is either Tensor or TensorList will be the deciding factor for dispatch.
If it is a TensorList, then that TensorList must not be empty or dispatch will fail.

Reviewed By: ezyang

Differential Revision: D14235840

fbshipit-source-id: 266c18912d56ce77aa84306c5605c4191f3d882b
aten/src/ATen/core/dispatch/DispatchTable.h