Searched refs:lrnK (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cudnn_6_0.inc | 1092 double lrnK ) {
1096 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
1104 double* lrnK ) {
1108 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
|
| D | cudnn_7_0.inc | 1204 double lrnK ) {
1208 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
1216 double* lrnK ) {
1220 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
|
| D | cudnn_7_3.inc | 1198 …iptor(cudnnLRNDescriptor_t normDesc, unsigned lrnN, double lrnAlpha, double lrnBeta, double lrnK) {
1202 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
1206 …r(cudnnLRNDescriptor_t normDesc, unsigned *lrnN, double *lrnAlpha, double *lrnBeta, double *lrnK) {
1210 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
|
| D | cudnn_7_1.inc | 1204 double lrnK ) {
1208 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
1216 double* lrnK ) {
1220 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
|
| D | cudnn_7_4.inc | 1198 …iptor(cudnnLRNDescriptor_t normDesc, unsigned lrnN, double lrnAlpha, double lrnBeta, double lrnK) {
1202 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
1206 …r(cudnnLRNDescriptor_t normDesc, unsigned *lrnN, double *lrnAlpha, double *lrnBeta, double *lrnK) {
1210 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
|
| D | cuda_dnn.cc | 781 double lrnK = normalize_descriptor.bias(); in CudnnNormalizeDescriptor() local
783 cudnnSetLRNDescriptor(handle_.get(), lrnN, lrnAlpha, lrnBeta, lrnK)); in CudnnNormalizeDescriptor()
|