theano/gpuarray/c_code · 7b13e2f49c91353e1306cfe68bd310a732af30db · testgroup / pytensor

multiple improvements to gpu topk · 7b13e2f4

由提交于 6月 25, 2017

 - added xlarge kernel to handle array size >= 2^31
 - ported original pytorch kernel
 - various small fixes

7b13e2f4

名称	最后提交	最后更新
..
blockgemv.c		正在载入提交数据...
blockger.c		正在载入提交数据...
conv_desc.c		正在载入提交数据...
corr3d_gemm.c		正在载入提交数据...
corr_gemm.c		正在载入提交数据...
ctc_wrapper.c		正在载入提交数据...
cudnn_helper.h		正在载入提交数据...
dimshuffle.c		正在载入提交数据...
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_redux.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		正在载入提交数据...
dnn_sptf_gi.c		正在载入提交数据...
dnn_sptf_grid.c		正在载入提交数据...
dnn_sptf_gt.c		正在载入提交数据...
dnn_sptf_sampler.c		正在载入提交数据...
gpuarray_helper.h		正在载入提交数据...
k_topk_common.cuh		正在载入提交数据...
k_topk_dense.cu		正在载入提交数据...
k_topk_dense_large.cu		正在载入提交数据...
magma_cholesky.c		正在载入提交数据...
magma_eigh.c		正在载入提交数据...
magma_inv.c		正在载入提交数据...
magma_qr.c		正在载入提交数据...
magma_svd.c		正在载入提交数据...
pool.c		正在载入提交数据...
pool_ave_grad.c		正在载入提交数据...
pool_grad_grad.c		正在载入提交数据...
pool_max_grad.c		正在载入提交数据...
pool_max_rop.c		正在载入提交数据...