| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| resample.py | 132 estimated_mean = variable_scope.get_local_variable(
133 'estimated_mean',
139 estimated_mean, batch_mean, mean_decay)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| fused_batch_norm_op.cc | 64 typename TTypes<U>::ConstVec estimated_mean(estimated_mean_input.vec<U>());
106 mean.device(d) = estimated_mean;
223 const Tensor& estimated_mean,
300 StreamExecutorUtil::AsDeviceMemory<U>(estimated_mean);
520 const Tensor& estimated_mean = context->input(3); variable
532 OP_REQUIRES(context, estimated_mean.dims() == 1,
533 errors::InvalidArgument("estimated_mean must be 1-dimensional",
534 estimated_mean.shape().DebugString()));
541 context, estimated_mean.dim_size(0) == 0,
542 errors::InvalidArgument("estimated_mean must be empty for training"
[all...] |
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_dnn.h | 203 const DeviceMemory<float>& estimated_mean,
216 const DeviceMemory<float>& estimated_mean,
[all...] |
| cuda_dnn.cc | [all...] |
| /external/tensorflow/tensorflow/stream_executor/ |
| dnn.h | 897 // estimated_mean: population mean estimated during training.
922 const DeviceMemory<float>& estimated_mean,
[all...] |
| stream.h | 224 const DeviceMemory<float> &estimated_mean,
245 const DeviceMemory<float> &estimated_mean,
[all...] |
| stream.cc | 317 const DeviceMemory<float> &estimated_mean,
331 this, x, scale, offset, estimated_mean, estimated_variance, x_desc,
367 const DeviceMemory<float> &estimated_mean,
381 this, x, scale, offset, estimated_mean, estimated_variance, x_desc,
[all...] |
