| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_eigen_output_kernels.h | 400 const Tensor& estimated_variance = context->input(5);
411 if (estimated_variance.dims() != 1)
413 estimated_variance.shape().DebugString());
422 args->estimated_variance_data = data_ptr(estimated_variance);
426 (estimated_variance.flat<T>() + static_cast<T>(epsilon)).rsqrt() *
|
| D | fused_batch_norm_op.cc | 299 typename TTypes<U>::ConstVec estimated_variance( in operator ()() local
323 auto scaling_factor = ((estimated_variance + epsilon).rsqrt() * scale) in operator ()()
334 batch_variance.device(d) = estimated_variance; in operator ()()
772 const Tensor& estimated_variance, const Tensor* side_input, in operator ()()
850 estimated_mean.vec<U>(), estimated_variance.vec<U>(), in operator ()()
857 estimated_mean.vec<U>(), estimated_variance.vec<U>(), in operator ()()
920 StreamExecutorUtil::AsDeviceMemory<U>(estimated_variance); in operator ()()
953 if (!batch_var->SharesBufferWith(estimated_variance) && in operator ()()
959 estimated_variance.NumElements() * sizeof(U)) in operator ()()
1191 typename TTypes<U>::ConstVec estimated_variance, \
[all …]
|
| D | fused_batch_norm_op.h | 49 typename TTypes<U>::ConstVec estimated_variance,
|
| D | fused_batch_norm_op.cu.cc | 322 typename TTypes<U>::ConstVec estimated_variance, in operator ()()
353 estimated_mean.data(), estimated_variance.data(), side_input.data(), \ in operator ()()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 287 const DeviceMemory<float>& estimated_variance,
301 const DeviceMemory<float>& estimated_variance,
646 const DeviceMemory<U>& estimated_variance,
|
| D | rocm_dnn.cc | 3640 const DeviceMemory<float>& estimated_variance, in DoBatchNormalizationForward() argument
3652 estimated_mean, estimated_variance, side_input, x_desc, scale_offset_desc, in DoBatchNormalizationForward()
3661 const DeviceMemory<float>& estimated_variance, in DoBatchNormalizationForward() argument
3672 estimated_mean, estimated_variance, side_input, x_desc, scale_offset_desc, in DoBatchNormalizationForward()
3683 const DeviceMemory<U>& estimated_variance, in DoBatchNormalizationForwardImpl() argument
3710 const void* maybe_inv_var = estimated_variance.opaque(); in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.h | 297 const DeviceMemory<float>& estimated_variance,
542 const DeviceMemory<U>& estimated_variance,
|
| D | cuda_dnn.cc | 5412 const DeviceMemory<float>& estimated_variance, in DoBatchNormalizationForward() argument
5424 offset, estimated_mean, estimated_variance, side_input, x_desc, in DoBatchNormalizationForward()
5435 const DeviceMemory<float>& estimated_variance, in DoBatchNormalizationForward() argument
5448 estimated_mean, estimated_variance, side_input, x_desc, in DoBatchNormalizationForward()
5461 const DeviceMemory<U>& estimated_variance, in DoBatchNormalizationForwardImpl() argument
5593 const void* maybe_inv_var = estimated_variance.opaque(); in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | stream.cc | 332 const DeviceMemory<float> &estimated_variance, in ThenBatchNormalizationForward() argument
345 this, x, scale, offset, estimated_mean, estimated_variance, side_input, in ThenBatchNormalizationForward()
385 const DeviceMemory<float> &estimated_variance, in ThenBatchNormalizationForward() argument
399 this, x, scale, offset, estimated_mean, estimated_variance, side_input, in ThenBatchNormalizationForward()
|
| D | dnn.h | 1166 const DeviceMemory<float>& estimated_variance, in DoBatchNormalizationForward() argument
1185 const DeviceMemory<float>& estimated_variance, in DoBatchNormalizationForward() argument
|
| D | stream.h | 256 const DeviceMemory<float> &estimated_variance,
282 const DeviceMemory<float> &estimated_variance,
|