tensorflow如何使用cudnn卷积?

时间:2018-01-24 09:01:54

标签: tensorflow cudnn

Tensorflow和cudnn卷积有不同的填充。但CPU / GPU卷积对用户来说是类似的。 tensorflow如何使用cudnn制作" tensoflow style"卷积?

1 个答案:

答案 0 :(得分:0)

我建议你看一下tf.nn.convolution的实现(在tensorflow / python / ops / nn_ops.py中),看看它是如何从python中完成的,以及在tensorflow / core / kernels /中实现conv2d内核conv_ops.cc用于C ++位(调用cudnn)。