| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 765 Tensor* batch_mean, Tensor* batch_var, Tensor* saved_mean, in operator ()()
918 auto saved_mean_ptr = StreamExecutorUtil::AsDeviceMemory<U>(*saved_mean); in operator ()()
1310 Tensor* saved_mean = nullptr; in ComputeWithReservedSpace() local
1312 context->allocate_output(3, scale.shape(), &saved_mean)); in ComputeWithReservedSpace()
1321 batch_mean, batch_var, saved_mean, saved_maybe_inv_var, in ComputeWithReservedSpace()
1327 batch_mean, batch_var, saved_mean, saved_maybe_inv_var, in ComputeWithReservedSpace()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1131 void* running_variance, void* saved_mean, void* saved_inv_variance, in SetBatchNormForwardArgs() argument
1142 fusion_args_, batchnorm_op, alpha, beta, scale, offset, saved_mean, in SetBatchNormForwardArgs()
1156 const void* saved_mean, in SetBatchNormBackwardArgs() argument
1168 offset_grad, saved_mean, saved_inv_variance); in SetBatchNormBackwardArgs()
1497 void* saved_mean, void* saved_var, in SetBatchNormForwardArgs() argument
1503 saved_mean, saved_var, epsilon, /*exponential_average_factor=*/1.0); in SetBatchNormForwardArgs()
1597 const void* saved_mean, in SetBatchNormBackwardArgs() argument
1605 offset_grad, saved_mean, saved_var); in SetBatchNormBackwardArgs()
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()
[all …]
|
| 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,
|
| /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
3612 /*resultSaveMean=*/saved_mean->opaque(), in DoBatchNormalizationForwardImpl()
3627 batch_mean_opaque, batch_var_opaque, epsilon, saved_mean->opaque(), in DoBatchNormalizationForwardImpl()
|
| /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()
|