Outputs of CUDNN depthwise conv mismatch Tensorflow depthwise conv

AI & Data Science Deep Learning (Training & Inference) cuDNN
1

The outputs of cudnn depthwise conv mismatched the outputs of tensorflow layer.
Is there anything wrong when I used cudnn?
Thanks a lot for your help!

conv_depthwise.cpp (6.1 KB)

2

Hi @user126573 ,
Can you please help us with the API log and code to repro on Tensorflow?
Thanks

3

conv.zip (13.9 KB)

Unzip the directory and run

python conv_tf.py

CUDNN codes are in conv_depthwise.cpp. Thanks a lot for your help!

4

Can anyone help me check the implementation?