pytorch/caffe2/python/helpers
James Reed f388135d3f Layer norm brew wrapper
Summary: Implement a brew wrapper for the LayerNorm op. This adds the scalar weight and bias terms to the op.

Reviewed By: jmp84

Differential Revision: D5595836

fbshipit-source-id: 467b2e1158b0c454a149d4b26c47719826e98752
2017-08-17 11:17:47 -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 Allow passing unsymmetric 2d kernels to brew.conv. 2017-08-10 15:27:16 -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 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