Cudnn Convolution BiasActivationForward problem

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

Hi, I am confused what the bias descriptor should be. It says it’s of type Tensor_t which suggests that it is in either NCHW or NHWC format. The bias just has C elements. Should I declare it in NCHW with 1s in the broadcast dimensions?

Is there a working example of this API usage?

2

Hi,

Bias should have the same dimension as input x. You can declare it in NCHW with 1s in the broadcast dimensions.
Please refer to below API link for supported datatypes and unsupported configurations.

For example you can refer to cuDNN Code Samples.

Thanks