• 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
名称
最后提交
最后更新
..
GpuArray.pkl 正在载入提交数据...
__init__.py 正在载入提交数据...
config.py 正在载入提交数据...
rnn_support.py 正在载入提交数据...
test_abstractconv.py 正在载入提交数据...
test_basic_ops.py 正在载入提交数据...
test_blas.py 正在载入提交数据...
test_blocksparse.py 正在载入提交数据...
test_cgpukernelbase.py 正在载入提交数据...
test_dnn.py 正在载入提交数据...
test_elemwise.py 正在载入提交数据...
test_extra_ops.py 正在载入提交数据...
test_fft.py 正在载入提交数据...
test_gemmcorr.py 正在载入提交数据...
test_gemmcorr3d.py 正在载入提交数据...
test_gpuarray_multinomial_wo_replacement.pkl 正在载入提交数据...
test_linalg.py 正在载入提交数据...
test_multinomial.py 正在载入提交数据...
test_neighbours.py 正在载入提交数据...
test_nerv.py 正在载入提交数据...
test_nnet.py 正在载入提交数据...
test_opt.py 正在载入提交数据...
test_others.py 正在载入提交数据...
test_pickle.py 正在载入提交数据...
test_pool.py 正在载入提交数据...
test_reduction.py 正在载入提交数据...
test_rng_mrg.py 正在载入提交数据...
test_scan.py 正在载入提交数据...
test_subtensor.py 正在载入提交数据...
test_type.py 正在载入提交数据...
tstgpueye.c 正在载入提交数据...