Fix half-float conversion ops to handle tensors larger than 2B of params (#17952)
authorAndrey Malevich <amalevich@fb.com>
Wed, 13 Mar 2019 05:57:44 +0000 (22:57 -0700)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Wed, 13 Mar 2019 06:03:22 +0000 (23:03 -0700)
commitc8f9072ab635e008ba88f4ad62478f8ac3379449
tree518e52f1d639add27f0c934b6b481ae6356cba1c
parent8bc3b66be9d448b3afb74aabc4c0dd826a1e00a5
Fix half-float conversion ops to handle tensors larger than 2B of params (#17952)

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

As desc.

Reviewed By: hyuen

Differential Revision: D14435092

fbshipit-source-id: dc614ba16ad531101d04d01aec8f1fbd534ebec5
caffe2/operators/half_float_ops.cc