Simon Layton 85788a0f65 Add TensorCore support ... Summary: Add support for TensorCore convolution and gemm on Volta hardware. Currently built on top of #1055 Closes https://github.com/caffe2/caffe2/pull/1056 Differential Revision: D5604068 Pulled By: Yangqing fbshipit-source-id: 100f67e26ed5fabb1dbb31dcd77f7ecb84de4ee7		2017-08-10 20:16: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	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	Option to set BN scale and bias initial values	2017-08-02 11:38:57 -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