| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 228 DeviceMemory<float>* saved_mean, DeviceMemory<float>* saved_inv_var,
242 DeviceMemory<float>* saved_mean, DeviceMemory<float>* saved_inv_var,
599 DeviceMemory<U>* saved_mean, DeviceMemory<U>* saved_inv_var,
|
| D | cuda_dnn.cc | 3457 DeviceMemory<float>* saved_mean, DeviceMemory<float>* saved_inv_var, in DoBatchNormalizationForward() argument
3465 activation_mode, y, batch_mean, batch_var, saved_mean, saved_inv_var, in DoBatchNormalizationForward()
3480 DeviceMemory<float>* saved_mean, DeviceMemory<float>* saved_inv_var, in DoBatchNormalizationForward() argument
3488 activation_mode, y, batch_mean, batch_var, saved_mean, saved_inv_var, in DoBatchNormalizationForward()
3505 DeviceMemory<U>* saved_mean, DeviceMemory<U>* saved_inv_var, in DoBatchNormalizationForwardImpl() argument
3613 /*resultSaveInvVariance=*/saved_inv_var->opaque(), in DoBatchNormalizationForwardImpl()
3628 saved_inv_var->opaque())); in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 266 DeviceMemory<float>* saved_mean, DeviceMemory<float>* saved_inv_var,
280 DeviceMemory<float>* saved_mean, DeviceMemory<float>* saved_inv_var,
715 DeviceMemory<U>* saved_mean, DeviceMemory<U>* saved_inv_var,
|
| D | rocm_dnn.cc | 3531 DeviceMemory<float>* saved_mean, DeviceMemory<float>* saved_inv_var, in DoBatchNormalizationForward() argument
3538 batch_var, saved_mean, saved_inv_var, is_training); in DoBatchNormalizationForward()
3551 DeviceMemory<float>* saved_mean, DeviceMemory<float>* saved_inv_var, in DoBatchNormalizationForward() argument
3558 batch_var, saved_mean, saved_inv_var, is_training); in DoBatchNormalizationForward()
3573 DeviceMemory<U>* saved_mean, DeviceMemory<U>* saved_inv_var, in DoBatchNormalizationForwardImpl() argument
3592 epsilon, saved_mean->opaque(), saved_inv_var->opaque()); in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 766 Tensor* saved_inv_var, TensorFormat tensor_format, in operator ()()
920 StreamExecutorUtil::AsDeviceMemory<U>(*saved_inv_var); in operator ()()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.h | 256 DeviceMemory<float> *saved_mean, DeviceMemory<float> *saved_inv_var,
281 DeviceMemory<float> *saved_mean, DeviceMemory<float> *saved_inv_var,
|
| D | stream.cc | 352 DeviceMemory<float> *saved_mean, DeviceMemory<float> *saved_inv_var, in ThenBatchNormalizationForward() argument
362 activation_mode, y, batch_mean, batch_var, saved_mean, saved_inv_var, in ThenBatchNormalizationForward()
403 DeviceMemory<float> *saved_mean, DeviceMemory<float> *saved_inv_var, in ThenBatchNormalizationForward() argument
413 activation_mode, y, batch_mean, batch_var, saved_mean, saved_inv_var, in ThenBatchNormalizationForward()
|