• notoraptor's avatar
    Wrap Op params for many gpuarray DNN Ops and add cuDNN v6 integration. · 560a4a56
    notoraptor 提交于
    Ops rewritten:
    
     - GpuDnnConvDesc
     - GpuDnnPool
     - GpuDnnPoolGrad
     - GpuDnnConv
     - GpuDnnConvGradW
     - GpuDnnConvGradI
     - GpuDnnBatchNormInference
     - GpuDnnBatchNormGrad
    
    cuDNN v6 integration:
    
     - Support MAX DETERMINISTIC algorithm for GpuDnnPool with cuDNN v6.
     - Update pooling tests for DNN module so that they use the right available algorithms depending on runtime cuDNN version.
     - Allow CPU Pool and PoolGrad ops to use MAX_DETERMINISTIC algo when cuDNN v6 is used with GPU counterparts.
     - Encapsulate cuDNN constants used in DNN module, to help choose the right cuDNN definitions depending on the runtime cuDNN version.
       Currently supported cuDNN versions: v5.1, v6.0.
    560a4a56
名称
最后提交
最后更新
..
tests 正在载入提交数据...
__init__.py 正在载入提交数据...
basic_ops.py 正在载入提交数据...
blas.py 正在载入提交数据...
blockgemv.c 正在载入提交数据...
blockger.c 正在载入提交数据...
blocksparse.py 正在载入提交数据...
conv_desc.c 正在载入提交数据...
corr3d_gemm.c 正在载入提交数据...
corr_gemm.c 正在载入提交数据...
cudnn_defs.py 正在载入提交数据...
cudnn_helper.h 正在载入提交数据...
dnn.py 正在载入提交数据...
dnn_base.c 正在载入提交数据...
dnn_batchnorm.c 正在载入提交数据...
dnn_batchnorm_base.c 正在载入提交数据...
dnn_batchnorm_grad.c 正在载入提交数据...
dnn_batchnorm_inf.c 正在载入提交数据...
dnn_conv_base.c 正在载入提交数据...
dnn_dropout_desc.c 正在载入提交数据...
dnn_dropout_fwd.c 正在载入提交数据...
dnn_fwd.c 正在载入提交数据...
dnn_gi.c 正在载入提交数据...
dnn_gw.c 正在载入提交数据...
dnn_pool.c 正在载入提交数据...
dnn_pool_grad.c 正在载入提交数据...
dnn_rnn_desc.c 正在载入提交数据...
dnn_rnn_fwd.c 正在载入提交数据...
dnn_rnn_gi.c 正在载入提交数据...
dnn_rnn_gw.c 正在载入提交数据...
dnn_rnn_paramsize.c 正在载入提交数据...
dnn_softmax.c 正在载入提交数据...
dnn_softmax_grad.c 正在载入提交数据...
elemwise.py 正在载入提交数据...
extra_ops.py 正在载入提交数据...
fft.py 正在载入提交数据...
fp16_help.py 正在载入提交数据...
gemm16.c 正在载入提交数据...
gpuarray_helper.h 正在载入提交数据...
kernel_codegen.py 正在载入提交数据...
linalg.py 正在载入提交数据...
magma_inv.c 正在载入提交数据...
magma_svd.c 正在载入提交数据...
multinomial.py 正在载入提交数据...
neighbours.py 正在载入提交数据...
nerv.py 正在载入提交数据...
nnet.py 正在载入提交数据...
opt.py 正在载入提交数据...
opt_util.py 正在载入提交数据...
pool.c 正在载入提交数据...
pool.py 正在载入提交数据...
pool_ave_grad.c 正在载入提交数据...
pool_grad_grad.c 正在载入提交数据...
pool_max_grad.c 正在载入提交数据...
pool_max_rop.c 正在载入提交数据...
reduction.py 正在载入提交数据...
rng_mrg.py 正在载入提交数据...
subtensor.py 正在载入提交数据...
type.py 正在载入提交数据...