pytorch/caffe2/python/helpers
Junjie Bai d9b0bcd7a4 Make all existing (except in RoIPool) "is_test" arguments required
Reviewed By: akyrola

Differential Revision: D5830168

fbshipit-source-id: 8634e9cfe308ba0ee90cd8a5c4b09a47b0b5f015
2017-09-25 23:46:12 -07:00
..
algebra.py resubmission of PR1175: fp16 BatchMatMul 2017-09-14 21:46:05 -07:00
arg_scope.py arg_scope for model_helper 2017-04-24 15:52:25 -07:00
array_helpers.py Bugfix for concat frontend 2017-09-11 13:02:59 -07:00
conv.py Allow passing unsymmetric 2d kernels to brew.conv. 2017-08-10 15:27:16 -07:00
dropout.py Make all existing (except in RoIPool) "is_test" arguments required 2017-09-25 23:46:12 -07:00
elementwise_linear.py add_helper_function_ElementwiseLinear_op 2017-06-07 13:49:48 -07:00
fc.py Add TensorCore support 2017-08-10 20:16:48 -07:00
nonlinearity.py Adding tanh to brew 2017-07-11 18:17:52 -07:00
normalization.py Layer norm brew wrapper 2017-08-17 11:17:47 -07:00
pooling.py New cudnn ops 2017-05-08 16:33:21 -07:00
tools.py Caffe2: Support additional outputs in ImageInputOp 2017-08-01 14:36:05 -07:00
train.py Adding add_weight_decay and image_input to brew module 2017-04-25 16:03:58 -07:00