Searched refs:lrnAlpha (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cudnn_6_0.inc | 1090 double lrnAlpha,
1096 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
1102 double* lrnAlpha,
1108 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
|
| D | cudnn_7_0.inc | 1202 double lrnAlpha,
1208 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
1214 double* lrnAlpha,
1220 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
|
| D | cudnn_7_3.inc | 1198 cudnnSetLRNDescriptor(cudnnLRNDescriptor_t normDesc, unsigned lrnN, double lrnAlpha, double lrnBeta…
1202 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
1206 cudnnGetLRNDescriptor(cudnnLRNDescriptor_t normDesc, unsigned *lrnN, double *lrnAlpha, double *lrnB…
1210 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
|
| D | cudnn_7_1.inc | 1202 double lrnAlpha,
1208 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
1214 double* lrnAlpha,
1220 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
|
| D | cudnn_7_4.inc | 1198 cudnnSetLRNDescriptor(cudnnLRNDescriptor_t normDesc, unsigned lrnN, double lrnAlpha, double lrnBeta…
1202 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
1206 cudnnGetLRNDescriptor(cudnnLRNDescriptor_t normDesc, unsigned *lrnN, double *lrnAlpha, double *lrnB…
1210 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
|
| D | cuda_dnn.cc | 778 double lrnAlpha = lrnN * normalize_descriptor.alpha(); in CudnnNormalizeDescriptor() local
783 cudnnSetLRNDescriptor(handle_.get(), lrnN, lrnAlpha, lrnBeta, lrnK)); in CudnnNormalizeDescriptor()
|