Searched refs:lrnAlpha (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cudnn_6_0.inc | 1109 unsigned lrnN, double lrnAlpha,
1115 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
1120 double *lrnAlpha,
1126 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
|
| D | cudnn_7_0.inc | 1234 unsigned lrnN, double lrnAlpha,
1240 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
1245 double *lrnAlpha,
1251 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
|
| D | cudnn_7_3.inc | 1234 unsigned lrnN, double lrnAlpha,
1240 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
1245 double *lrnAlpha,
1251 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
|
| D | cudnn_7_1.inc | 1234 unsigned lrnN, double lrnAlpha,
1240 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
1245 double *lrnAlpha,
1251 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
|
| D | cudnn_7_4.inc | 1234 unsigned lrnN, double lrnAlpha,
1240 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
1245 double *lrnAlpha,
1251 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
|
| D | cudnn_7_6.inc | 1399 unsigned lrnN, double lrnAlpha,
1405 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
1410 double *lrnAlpha,
1416 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
|
| D | cudnn_8_0.inc | 748 unsigned lrnN, double lrnAlpha,
754 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
759 double *lrnAlpha,
765 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
|
| D | cuda_dnn.cc | 996 double lrnAlpha = lrnN * normalize_descriptor.alpha(); in CudnnNormalizeDescriptor() local
1001 cudnnSetLRNDescriptor(handle_.get(), lrnN, lrnAlpha, lrnBeta, lrnK)); in CudnnNormalizeDescriptor()
|