Simon Layton 1d0ba2cfbd New cudnn ops ... Summary: cuDNN versions of dropout and LRN (for native fp16 support), port of Caffe's max pooling algo that uses an explicit mask to store locations (also supports fp16 storage) Closes https://github.com/caffe2/caffe2/pull/396 Reviewed By: akyrola Differential Revision: D4990880 Pulled By: asaadaldien fbshipit-source-id: a716acffb656843e9b31e3e6808bd2d8aa959d03		2017-05-08 16:33:21 -07:00
..
algebra.py	rename model_helpers to brew and lowercase all helper functions	2017-04-24 15:52:26 -07:00
arg_scope.py	arg_scope for model_helper	2017-04-24 15:52:25 -07:00
array_helpers.py	rename model_helpers to brew and lowercase all helper functions	2017-04-24 15:52:26 -07:00
conv.py	rename model_helpers to brew and lowercase all helper functions	2017-04-24 15:52:26 -07:00
dropout.py	New cudnn ops	2017-05-08 16:33:21 -07:00
fc.py	rename model_helpers to brew and lowercase all helper functions	2017-04-24 15:52:26 -07:00
nonlinearity.py	rename model_helpers to brew and lowercase all helper functions	2017-04-24 15:52:26 -07:00
normalization.py	New cudnn ops	2017-05-08 16:33:21 -07:00
pooling.py	New cudnn ops	2017-05-08 16:33:21 -07:00
tools.py	Adding video data layer for caffe2	2017-05-05 14:16:38 -07:00
train.py	Adding add_weight_decay and image_input to brew module	2017-04-25 16:03:58 -07:00