pytorch/test/cpp
Mansoor e95657b87e [C++ API] AdaptiveLogSoftmaxWithLoss (#29076)
Summary:
Implemented AdaptiveLogSoftmaxWithLoss and some tests for modules. Reference https://github.com/pytorch/pytorch/issues/25883
Pull Request resolved: https://github.com/pytorch/pytorch/pull/29076

Differential Revision: D20404588

Pulled By: yf225

fbshipit-source-id: edbadf432b8173cbcc6caf83c9c03dd92dc31a37
2020-03-12 09:53:58 -07:00
..
api [C++ API] AdaptiveLogSoftmaxWithLoss (#29076) 2020-03-12 09:53:58 -07:00
common Trim libshm deps, move tempfile.h to c10 (#17019) 2019-02-13 19:38:35 -08:00
dist_autograd [Dist Autograd] Functional API for Dist Autograd and Dist Optimizer (#33711) 2020-02-26 19:08:28 -08:00
jit [jit] kill script namespace (#34515) 2020-03-11 23:32:48 -07:00
rpc Remove dead includes in caffe2/test 2020-01-21 11:30:34 -08:00
tensorexpr [TensorExpr] Add IR Printer. (#33220) 2020-02-21 13:10:26 -08:00
__init__.py Add train() / eval() / is_training() to C++ ScriptModule API (#16044) 2019-02-01 13:07:38 -08:00