Searched refs:lrnK (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cudnn_6_0.inc | 1110 double lrnBeta, double lrnK) {
1115 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
1121 double *lrnBeta, double *lrnK) {
1126 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
|
| D | cudnn_7_0.inc | 1235 double lrnBeta, double lrnK) {
1240 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
1246 double *lrnBeta, double *lrnK) {
1251 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
|
| D | cudnn_7_3.inc | 1235 double lrnBeta, double lrnK) {
1240 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
1246 double *lrnBeta, double *lrnK) {
1251 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
|
| D | cudnn_7_1.inc | 1235 double lrnBeta, double lrnK) {
1240 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
1246 double *lrnBeta, double *lrnK) {
1251 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
|
| D | cudnn_7_4.inc | 1235 double lrnBeta, double lrnK) {
1240 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
1246 double *lrnBeta, double *lrnK) {
1251 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
|
| D | cudnn_8_0.inc | 749 double lrnBeta, double lrnK) {
754 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
760 double *lrnBeta, double *lrnK) {
765 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
|
| D | cudnn_7_6.inc | 1400 double lrnBeta, double lrnK) {
1405 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
1411 double *lrnBeta, double *lrnK) {
1416 return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK);
|
| D | cuda_dnn.cc | 842 double lrnK = normalize_descriptor.bias(); in CudnnNormalizeDescriptor() local
844 cudnnSetLRNDescriptor(handle_.get(), lrnN, lrnAlpha, lrnBeta, lrnK)); in CudnnNormalizeDescriptor()
|