Commit Graph

17 Commits

Author SHA1 Message Date
Philippe Tillet
d9945692a9 [dnn] better specification of recompilation key 2019-08-02 17:42:48 -07:00
Philippe Tillet
5af7e5adac Made sure it works for FP16 2019-07-30 20:02:16 -07:00
Philippe Tillet
17cb2db356 [dnn/blocksparse/dot] prototype version seems to pass basic test 2019-07-27 21:21:36 -07:00
Philippe Tillet
b1d81a5802 more work on heuristics 2019-07-21 18:11:54 -07:00
Philippe Tillet
07c964919c [dnn/shift] now strictly only shifting the interior 2019-07-16 20:18:48 -07:00
Philippe Tillet
3e7a3ed67a [dnn/shift]: added support for fp16 2019-07-13 21:05:34 -07:00
Philippe Tillet
f36a646ffc [dnn/shift-conv] added and tested NCHW layout 2019-07-11 21:00:33 -07:00
Philippe Tillet
63b249c1d6 [examples/python/pytorch] added batchnorm cpp extension 2019-07-09 20:59:04 -07:00
Philippe Tillet
b7986baffa [dnn]: Now implementing all existing DNN routines using common base template and auto-tuner 2019-07-09 19:52:55 -07:00
Philippe Tillet
88675fa01a [dnn] added base template class for mutualized auto-tuning 2019-07-09 16:09:34 -07:00
Philippe Tillet
066ae338f1 [dnn/shift]: added stride to shift 2019-07-09 14:08:51 -07:00
Philippe Tillet
f9db0449b7 [dnn] Adding batchnorm 2019-07-08 18:44:37 -07:00
Philippe Tillet
b0cf3143c5 [dnn/shift] bugfix in wgrad 2019-07-06 11:27:49 -07:00
Philippe Tillet
3e49dbe6ab [dnn/shift] fixed in leading dimensions for shift-conv operation 2019-07-05 17:17:22 -07:00
Philippe Tillet
1d88f0a36b stuff 2019-07-03 19:25:16 -07:00
Philippe Tillet
0d8faa5b1e fixup 2019-07-02 21:38:10 -07:00
Philippe Tillet
5144dc3a6c [examples/python] added framework code for shift-conv 2019-07-02 20:45:10 -07:00