pytorch/caffe2/python/helpers
Aapo Kyrola fb45383ed6 resubmission of PR1175: fp16 BatchMatMul
Summary: PR 1175 caused a build error because gemmBatched was only under a specific #ifdef. Now put it outside the #ifdef, and things work.

Reviewed By: asaadaldien

Differential Revision: D5834868

fbshipit-source-id: 072a64c8f4b259ff7504104121766115b46b8aa0
2017-09-14 21:46:05 -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 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