Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc895 unsigned lrn_N = 2 * normalize_descriptor.range() + 1; in ScopedNormalizeDescriptor() local
907 double lrn_alpha = lrn_N * normalize_descriptor.alpha(); in ScopedNormalizeDescriptor()
911 status = wrap::miopenSetLRNDescriptor(handle_, miopenLRNCrossChannel, lrn_N, in ScopedNormalizeDescriptor()