trunk,cuda: fixing CuMatrix::AddMat(.) to use 2strides; adding...
trunk,cuda: fixing CuMatrix::AddMat(.) to use 2strides; adding CuMatrix::EqualElementMask(.) which produces per-element equality mask (handy for backpropagation through max-pooling) git-svn-id: https://svn.code.sf.net/p/kaldi/code/trunk@3400 5e6a8d80-dfce-4ca6-a32a-6e07a63d50c8
Showing
- src/cudamatrix/cu-kernels-ansi.h 9 additions, 3 deletionssrc/cudamatrix/cu-kernels-ansi.h
- src/cudamatrix/cu-kernels.cu 31 additions, 7 deletionssrc/cudamatrix/cu-kernels.cu
- src/cudamatrix/cu-kernels.h 14 additions, 2 deletionssrc/cudamatrix/cu-kernels.h
- src/cudamatrix/cu-matrix-test.cc 29 additions, 0 deletionssrc/cudamatrix/cu-matrix-test.cc
- src/cudamatrix/cu-matrix.cc 31 additions, 1 deletionsrc/cudamatrix/cu-matrix.cc
- src/cudamatrix/cu-matrix.h 5 additions, 0 deletionssrc/cudamatrix/cu-matrix.h
Loading
Please register or sign in to comment