| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 81 typename TTypes<U>::ConstVec estimated_mean(estimated_mean_input.vec<U>()); in operator ()() local
123 mean.device(d) = estimated_mean; in operator ()()
244 const Tensor& estimated_mean, in operator ()()
321 StreamExecutorUtil::AsDeviceMemory<U>(estimated_mean); in operator ()()
541 const Tensor& estimated_mean = context->input(3); in Compute() local
553 OP_REQUIRES(context, estimated_mean.dims() == 1, in Compute()
555 estimated_mean.shape().DebugString())); in Compute()
562 context, estimated_mean.dim_size(0) == 0, in Compute()
564 estimated_mean.shape().DebugString())); in Compute()
588 context, x, scale, offset, estimated_mean, estimated_variance, epsilon_, in Compute()
|
| D | conv_ops_fused_impl.h | 409 const Tensor& estimated_mean = context->input(4);
418 TF_REQUIRES(estimated_mean.dims() == 1,
420 estimated_mean.shape().DebugString()));
428 InitDataPtr(estimated_mean, &args->estimated_mean_data);
|
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| D | resample.py | 132 estimated_mean = variable_scope.get_local_variable(
139 estimated_mean, batch_mean, mean_decay)
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 220 const DeviceMemory<float>& estimated_mean,
233 const DeviceMemory<float>& estimated_mean,
561 const DeviceMemory<U>& estimated_mean,
|
| D | cuda_dnn.cc | 3181 const DeviceMemory<float>& estimated_mean, in DoBatchNormalizationForward() argument
3193 offset, estimated_mean, estimated_variance, x_desc, scale_offset_desc, in DoBatchNormalizationForward()
3202 const DeviceMemory<float>& estimated_mean, in DoBatchNormalizationForward() argument
3214 estimated_mean, estimated_variance, x_desc, scale_offset_desc, in DoBatchNormalizationForward()
3225 const DeviceMemory<U>& estimated_mean, in DoBatchNormalizationForwardImpl() argument
3273 scale.opaque(), offset.opaque(), estimated_mean.opaque(), maybe_inv_var, in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 211 const DeviceMemory<float>& estimated_mean,
224 const DeviceMemory<float>& estimated_mean,
633 const DeviceMemory<U>& estimated_mean,
|
| D | rocm_dnn.cc | 2999 const DeviceMemory<float>& estimated_mean, in DoBatchNormalizationForward() argument
3010 estimated_mean, estimated_variance, x_desc, scale_offset_desc, epsilon, y, in DoBatchNormalizationForward()
3018 const DeviceMemory<float>& estimated_mean, in DoBatchNormalizationForward() argument
3029 estimated_mean, estimated_variance, x_desc, scale_offset_desc, epsilon, y, in DoBatchNormalizationForward()
3039 const DeviceMemory<U>& estimated_mean, in DoBatchNormalizationForwardImpl() argument
3073 const_cast<void*>(estimated_mean.opaque()), in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 993 const DeviceMemory<float>& estimated_mean,
1010 const DeviceMemory<float>& estimated_mean,
|
| D | stream.h | 235 const DeviceMemory<float> &estimated_mean,
256 const DeviceMemory<float> &estimated_mean,
|
| D | stream.cc | 342 const DeviceMemory<float> &estimated_mean, in ThenBatchNormalizationForward() argument
356 this, x, scale, offset, estimated_mean, estimated_variance, x_desc, in ThenBatchNormalizationForward()
392 const DeviceMemory<float> &estimated_mean, in ThenBatchNormalizationForward() argument
406 this, x, scale, offset, estimated_mean, estimated_variance, x_desc, in ThenBatchNormalizationForward()
|