pytorch/test/cpp
Thomas J. Fan a7f4f80903 ENH Adds dtype to nn.functional.one_hot (#58090)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/33046
Related to https://github.com/pytorch/pytorch/issues/53785

Pull Request resolved: https://github.com/pytorch/pytorch/pull/58090

Reviewed By: zou3519

Differential Revision: D28640893

Pulled By: jbschlosser

fbshipit-source-id: 3686579517ccc75beaa74f0f6d167f5e40a83fd2
2021-05-24 13:48:25 -07:00
..
api ENH Adds dtype to nn.functional.one_hot (#58090) 2021-05-24 13:48:25 -07:00
common
dist_autograd Fix distributed autograd gradients synchronization (#57792) 2021-05-09 17:32:59 -07:00
jit [Pytorch Edge] Model Ops compatibility api (#57501) 2021-05-24 12:00:06 -07:00
lite_interpreter_runtime [Pytorch Backend delegation] Add api for backend lowering to query debug (#55462) 2021-05-22 08:33:07 -07:00
rpc Revert D28474878: Always use intrusive_ptr for Message (1 out of 2) 2021-05-21 19:24:22 -07:00
tensorexpr [NNC] Do not optimize conditionals when the corresponding loop is not normalized (#57675) 2021-05-18 14:25:53 -07:00
__init__.py