fix wrong schema (#19370)
authorElias Ellison <eellison@fb.com>
Thu, 18 Apr 2019 02:52:16 +0000 (19:52 -0700)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Thu, 18 Apr 2019 02:55:30 +0000 (19:55 -0700)
commit776fec0f9f3ea5ebd7cdd18d52e8702d81064bc7
tree17c3a0ada0d8e144b6c542c8711ea77eb68fec1f
parentbf5f30f39bc80cdf7f1e0770a63189b48bd9209b
fix wrong schema (#19370)

Summary:
Op was improperly schematized previously.  Evidently checkScript does not test if the outputs are the same type.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/19370

Differential Revision: D14985159

Pulled By: eellison

fbshipit-source-id: feb60552afa2a6956d71f64801f15e5fe19c3a91
test/test_jit.py
torch/csrc/jit/register_prim_ops.cpp