| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_eigen_output_kernels.h | 399 const Tensor& estimated_variance = context->input(5);
410 if (estimated_variance.dims() != 1)
412 estimated_variance.shape().DebugString());
421 args->estimated_variance_data = data_ptr(estimated_variance);
425 (estimated_variance.flat<T>() + static_cast<T>(epsilon)).rsqrt() *
|
| D | fused_batch_norm_op.cc | 287 typename TTypes<U>::ConstVec estimated_variance( in operator ()() local
315 auto scaling_factor = ((estimated_variance + epsilon).rsqrt() * scale) in operator ()()
326 batch_variance.device(d) = estimated_variance; in operator ()()
762 const Tensor& estimated_variance, const Tensor* side_input, in operator ()()
840 estimated_mean.vec<U>(), estimated_variance.vec<U>(), in operator ()()
847 estimated_mean.vec<U>(), estimated_variance.vec<U>(), in operator ()()
910 StreamExecutorUtil::AsDeviceMemory<U>(estimated_variance); in operator ()()
943 if (!batch_var->SharesBufferWith(estimated_variance) && in operator ()()
949 estimated_variance.NumElements() * sizeof(U)) in operator ()()
1165 typename TTypes<U>::ConstVec estimated_variance, \
[all …]
|
| D | fused_batch_norm_op.h | 55 typename TTypes<U>::ConstVec estimated_variance,
|
| D | fused_batch_norm_op.cu.cc | 324 typename TTypes<U>::ConstVec estimated_variance, in operator ()()
355 estimated_mean.data(), estimated_variance.data(), side_input.data(), \ in operator ()()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 260 const DeviceMemory<float>& estimated_variance,
274 const DeviceMemory<float>& estimated_variance,
709 const DeviceMemory<U>& estimated_variance,
|
| D | rocm_dnn.cc | 3525 const DeviceMemory<float>& estimated_variance, in DoBatchNormalizationForward() argument
3536 estimated_mean, estimated_variance, side_input, x_desc, scale_offset_desc, in DoBatchNormalizationForward()
3545 const DeviceMemory<float>& estimated_variance, in DoBatchNormalizationForward() argument
3556 estimated_mean, estimated_variance, side_input, x_desc, scale_offset_desc, in DoBatchNormalizationForward()
3567 const DeviceMemory<U>& estimated_variance, in DoBatchNormalizationForwardImpl() argument
3594 const void* maybe_inv_var = estimated_variance.opaque(); in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 236 const DeviceMemory<float>& estimated_variance,
593 const DeviceMemory<U>& estimated_variance,
|
| D | cuda_dnn.cc | 3451 const DeviceMemory<float>& estimated_variance, in DoBatchNormalizationForward() argument
3463 offset, estimated_mean, estimated_variance, side_input, x_desc, in DoBatchNormalizationForward()
3474 const DeviceMemory<float>& estimated_variance, in DoBatchNormalizationForward() argument
3486 estimated_mean, estimated_variance, side_input, x_desc, in DoBatchNormalizationForward()
3499 const DeviceMemory<U>& estimated_variance, in DoBatchNormalizationForwardImpl() argument
3631 const void* maybe_inv_var = estimated_variance.opaque(); in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 1015 const DeviceMemory<float>& estimated_variance, in DoBatchNormalizationForward() argument
1034 const DeviceMemory<float>& estimated_variance, in DoBatchNormalizationForward() argument
|
| D | stream.h | 250 const DeviceMemory<float> &estimated_variance,
275 const DeviceMemory<float> &estimated_variance,
|
| D | stream.cc | 346 const DeviceMemory<float> &estimated_variance, in ThenBatchNormalizationForward() argument
360 this, x, scale, offset, estimated_mean, estimated_variance, side_input, in ThenBatchNormalizationForward()
397 const DeviceMemory<float> &estimated_variance, in ThenBatchNormalizationForward() argument
411 this, x, scale, offset, estimated_mean, estimated_variance, side_input, in ThenBatchNormalizationForward()
|