Searched defs:saved_inv_var (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 3187 DeviceMemory<float>* saved_inv_var, bool is_training, in DoBatchNormalizationForward()
3208 DeviceMemory<float>* saved_inv_var, bool is_training, in DoBatchNormalizationForward()
3230 DeviceMemory<U>* saved_mean, DeviceMemory<U>* saved_inv_var, in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3005 DeviceMemory<float>* saved_inv_var, bool is_training, in DoBatchNormalizationForward()
3024 DeviceMemory<float>* saved_inv_var, bool is_training, in DoBatchNormalizationForward()
3044 DeviceMemory<U>* saved_mean, DeviceMemory<U>* saved_inv_var, in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 348 DeviceMemory<float> *saved_inv_var, bool is_training, in ThenBatchNormalizationForward()
398 DeviceMemory<float> *saved_inv_var, bool is_training, in ThenBatchNormalizationForward()
|