cavis/libnd4j/include/ops/declarable/platform/cudnn
agibsonccc 18d165c915 Update cuda compute defaults and ctc loss 2021-03-12 18:58:45 +09:00
..
avgpool2d.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
avgpool3d.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
batchnorm.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
conv2d.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
conv3d.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
ctcloss.cu Update cuda compute defaults and ctc loss 2021-03-12 18:58:45 +09:00
cudnnUtils.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
cudnnUtils.h Add ctc loss from KonduitAI PR, add missing java bits 2021-03-11 14:22:34 +09:00
depthwiseConv2d.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
maxpool2d.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
maxpool3d.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00