Searched defs:saved_inv_var (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 3457 DeviceMemory<float>* saved_mean, DeviceMemory<float>* saved_inv_var, in DoBatchNormalizationForward()
3480 DeviceMemory<float>* saved_mean, DeviceMemory<float>* saved_inv_var, in DoBatchNormalizationForward()
3505 DeviceMemory<U>* saved_mean, DeviceMemory<U>* saved_inv_var, in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3531 DeviceMemory<float>* saved_mean, DeviceMemory<float>* saved_inv_var, in DoBatchNormalizationForward()
3551 DeviceMemory<float>* saved_mean, DeviceMemory<float>* saved_inv_var, in DoBatchNormalizationForward()
3573 DeviceMemory<U>* saved_mean, DeviceMemory<U>* saved_inv_var, in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 352 DeviceMemory<float> *saved_mean, DeviceMemory<float> *saved_inv_var, in ThenBatchNormalizationForward()
403 DeviceMemory<float> *saved_mean, DeviceMemory<float> *saved_inv_var, in ThenBatchNormalizationForward()
|