Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1013 const void* offset, void* running_mean, in SetBatchNormForwardArgs() argument
1030 saved_inv_variance, running_mean, running_variance, exp_avg_factor, in SetBatchNormForwardArgs()