| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 82 typename TTypes<U>::ConstVec estimated_variance( in operator ()() local
134 variance.device(d) = estimated_variance; in operator ()()
245 const Tensor& estimated_variance, U epsilon, Tensor* y, in operator ()()
323 StreamExecutorUtil::AsDeviceMemory<U>(estimated_variance); in operator ()()
336 estimated_variance.shape(), &inv_var)); in operator ()()
339 [d, epsilon, estimated_variance, in operator ()()
342 StreamExecutorUtil::AsDeviceMemory<U>(estimated_variance); in operator ()()
542 const Tensor& estimated_variance = context->input(4); in Compute() local
557 context, estimated_variance.dims() == 1, in Compute()
559 estimated_variance.shape().DebugString())); in Compute()
[all …]
|
| D | conv_ops_fused_impl.h | 410 const Tensor& estimated_variance = context->input(5);
422 estimated_variance.dims() == 1,
424 estimated_variance.shape().DebugString()));
429 InitDataPtr(estimated_variance, &args->estimated_variance_data);
433 (estimated_variance.flat<T>() + static_cast<T>(epsilon)).rsqrt() *
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 221 const DeviceMemory<float>& estimated_variance,
234 const DeviceMemory<float>& estimated_variance,
562 const DeviceMemory<U>& estimated_variance,
|
| D | cuda_dnn.cc | 3182 const DeviceMemory<float>& estimated_variance, in DoBatchNormalizationForward() argument
3193 offset, estimated_mean, estimated_variance, x_desc, scale_offset_desc, in DoBatchNormalizationForward()
3203 const DeviceMemory<float>& estimated_variance, in DoBatchNormalizationForward() argument
3214 estimated_mean, estimated_variance, x_desc, scale_offset_desc, in DoBatchNormalizationForward()
3226 const DeviceMemory<U>& estimated_variance, in DoBatchNormalizationForwardImpl() argument
3269 const void* maybe_inv_var = estimated_variance.opaque(); in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 212 const DeviceMemory<float>& estimated_variance,
225 const DeviceMemory<float>& estimated_variance,
634 const DeviceMemory<U>& estimated_variance,
|
| D | rocm_dnn.cc | 3000 const DeviceMemory<float>& estimated_variance, in DoBatchNormalizationForward() argument
3010 estimated_mean, estimated_variance, x_desc, scale_offset_desc, epsilon, y, in DoBatchNormalizationForward()
3019 const DeviceMemory<float>& estimated_variance, in DoBatchNormalizationForward() argument
3029 estimated_mean, estimated_variance, x_desc, scale_offset_desc, epsilon, y, in DoBatchNormalizationForward()
3040 const DeviceMemory<U>& estimated_variance, in DoBatchNormalizationForwardImpl() argument
3068 const void* maybe_inv_var = estimated_variance.opaque(); in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 994 const DeviceMemory<float>& estimated_variance,
1011 const DeviceMemory<float>& estimated_variance,
|
| D | stream.h | 236 const DeviceMemory<float> &estimated_variance,
257 const DeviceMemory<float> &estimated_variance,
|
| D | stream.cc | 343 const DeviceMemory<float> &estimated_variance, in ThenBatchNormalizationForward() argument
356 this, x, scale, offset, estimated_mean, estimated_variance, x_desc, in ThenBatchNormalizationForward()
393 const DeviceMemory<float> &estimated_variance, in ThenBatchNormalizationForward() argument
406 this, x, scale, offset, estimated_mean, estimated_variance, x_desc, in ThenBatchNormalizationForward()
|