Added assisted_excitation=1 for [convolutional] layer on GPU

This commit is contained in:
AlexeyAB
2019-09-03 01:35:05 +03:00
parent a63782ca89
commit be5d0d6693
8 changed files with 232 additions and 26 deletions

View File

@ -537,6 +537,9 @@ struct layer {
float * rand_gpu;
float * squared_gpu;
float * norms_gpu;
float *gt_gpu;
float *a_avg_gpu;
#ifdef CUDNN
cudnnTensorDescriptor_t srcTensorDesc, dstTensorDesc;
cudnnTensorDescriptor_t srcTensorDesc16, dstTensorDesc16;