Home
last modified time | relevance | path

Searched refs:lrnN (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcudnn_6_0.inc1109 unsigned lrnN, double lrnAlpha,
1115 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
1119 unsigned *lrnN,
1126 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
Dcudnn_7_0.inc1234 unsigned lrnN, double lrnAlpha,
1240 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
1244 unsigned *lrnN,
1251 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
Dcudnn_7_3.inc1234 unsigned lrnN, double lrnAlpha,
1240 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
1244 unsigned *lrnN,
1251 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
Dcudnn_7_1.inc1234 unsigned lrnN, double lrnAlpha,
1240 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
1244 unsigned *lrnN,
1251 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
Dcudnn_7_4.inc1234 unsigned lrnN, double lrnAlpha,
1240 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
1244 unsigned *lrnN,
1251 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
Dcuda_dnn.cc827 unsigned lrnN = 2 * normalize_descriptor.range() + 1; in CudnnNormalizeDescriptor() local
839 double lrnAlpha = lrnN * normalize_descriptor.alpha(); in CudnnNormalizeDescriptor()
844 cudnnSetLRNDescriptor(handle_.get(), lrnN, lrnAlpha, lrnBeta, lrnK)); in CudnnNormalizeDescriptor()
Dcudnn_8_0.inc748 unsigned lrnN, double lrnAlpha,
754 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
758 unsigned *lrnN,
765 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
Dcudnn_7_6.inc1399 unsigned lrnN, double lrnAlpha,
1405 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
1409 unsigned *lrnN,
1416 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);