pytorch/caffe2/python/helpers
Luke Yeager dfd745a4d1 Conv frontend: checking engine and use_cudnn
Summary:
*Fixes https://github.com/caffe2/caffe2/issues/860*

Raise an exception when the user specifies conflicting values for `engine` and `use_cudnn` in the conv frontend.
Closes https://github.com/caffe2/caffe2/pull/861

Differential Revision: D5329587

Pulled By: akyrola

fbshipit-source-id: 0f1ced9a88c9c6c5a7cb30a070e5bf60129082f0
2017-06-27 09:47:48 -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 Conv frontend: checking engine and use_cudnn 2017-06-27 09:47:48 -07:00
dropout.py New cudnn ops 2017-05-08 16:33:21 -07:00
elementwise_linear.py add_helper_function_ElementwiseLinear_op 2017-06-07 13:49:48 -07:00
fc.py Create ExternalInitializer to simplify logic around init_params = False 2017-06-02 18:22:50 -07:00
nonlinearity.py Refactoring of the parameters step 0. Add simple tags and unify interface for params and computed_params. 2017-06-02 17:17:57 -07:00
normalization.py Refactoring of the parameters step 0. Add simple tags and unify interface for params and computed_params. 2017-06-02 17:17:57 -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