| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 87 typename TTypes<U>::Vec saved_mean(saved_mean_output->vec<U>()); in operator ()() local
121 saved_mean.device(d) = mean; in operator ()()
246 Tensor* batch_mean, Tensor* batch_var, Tensor* saved_mean, in operator ()()
327 auto saved_mean_ptr = StreamExecutorUtil::AsDeviceMemory<U>(*saved_mean); in operator ()()
580 Tensor* saved_mean = nullptr; in Compute() local
582 context->allocate_output(3, scale.shape(), &saved_mean)); in Compute()
589 y, batch_mean, batch_var, saved_mean, saved_maybe_inv_var, in Compute()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1015 void* saved_mean, in SetBatchNormForwardArgs() argument
1029 fusion_args_, batchnorm_op, alpha, beta, scale, offset, saved_mean, in SetBatchNormForwardArgs()
1043 const void* saved_mean, in SetBatchNormBackwardArgs() argument
1055 offset_grad, saved_mean, saved_inv_variance); in SetBatchNormBackwardArgs()
1384 void* saved_mean, void* saved_var, in SetBatchNormForwardArgs() argument
1390 saved_mean, saved_var, epsilon); in SetBatchNormForwardArgs()
1484 const void* saved_mean, in SetBatchNormBackwardArgs() argument
1492 offset_grad, saved_mean, saved_var); in SetBatchNormBackwardArgs()
3004 DeviceMemory<float>* batch_var, DeviceMemory<float>* saved_mean, in DoBatchNormalizationForward() argument
3011 batch_mean, batch_var, saved_mean, saved_inv_var, is_training, in DoBatchNormalizationForward()
[all …]
|
| D | rocm_dnn.h | 216 DeviceMemory<float>* batch_var, DeviceMemory<float>* saved_mean,
229 DeviceMemory<float>* batch_var, DeviceMemory<float>* saved_mean,
638 DeviceMemory<U>* batch_var, DeviceMemory<U>* saved_mean,
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 225 DeviceMemory<float>* batch_var, DeviceMemory<float>* saved_mean,
238 DeviceMemory<float>* batch_var, DeviceMemory<float>* saved_mean,
566 DeviceMemory<U>* batch_var, DeviceMemory<U>* saved_mean,
|
| D | cuda_dnn.cc | 3186 DeviceMemory<float>* batch_var, DeviceMemory<float>* saved_mean, in DoBatchNormalizationForward() argument
3194 epsilon, y, batch_mean, batch_var, saved_mean, saved_inv_var, in DoBatchNormalizationForward()
3207 DeviceMemory<float>* batch_var, DeviceMemory<float>* saved_mean, in DoBatchNormalizationForward() argument
3215 epsilon, y, batch_mean, batch_var, saved_mean, saved_inv_var, in DoBatchNormalizationForward()
3230 DeviceMemory<U>* saved_mean, DeviceMemory<U>* saved_inv_var, in DoBatchNormalizationForwardImpl() argument
3266 batch_var_opaque, epsilon, saved_mean->opaque(), in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.h | 240 DeviceMemory<float> *batch_var, DeviceMemory<float> *saved_mean,
261 DeviceMemory<float> *batch_var, DeviceMemory<float> *saved_mean,
|
| D | stream.cc | 347 DeviceMemory<float> *batch_var, DeviceMemory<float> *saved_mean, in ThenBatchNormalizationForward() argument
357 scale_offset_desc, epsilon, y, batch_mean, batch_var, saved_mean, in ThenBatchNormalizationForward()
397 DeviceMemory<float> *batch_var, DeviceMemory<float> *saved_mean, in ThenBatchNormalizationForward() argument
407 scale_offset_desc, epsilon, y, batch_mean, batch_var, saved_mean, in ThenBatchNormalizationForward()
|