Searched defs:saved_mean (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1131 void* running_variance, void* saved_mean, void* saved_inv_variance, in SetBatchNormForwardArgs()
1156 const void* saved_mean, in SetBatchNormBackwardArgs()
1497 void* saved_mean, void* saved_var, in SetBatchNormForwardArgs()
1597 const void* saved_mean, in SetBatchNormBackwardArgs()
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/core/kernels/ |
| D | fused_batch_norm_op.cc | 1310 Tensor* saved_mean = nullptr; in ComputeWithReservedSpace() local
|
| /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/ |
| 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()
|