| /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 | 81 typename TTypes<U>::ConstVec estimated_mean(estimated_mean_input.vec<U>());
123 mean.device(d) = estimated_mean;
244 const Tensor& estimated_mean,
321 StreamExecutorUtil::AsDeviceMemory<U>(estimated_mean);
541 const Tensor& estimated_mean = context->input(3); variable
553 OP_REQUIRES(context, estimated_mean.dims() == 1,
554 errors::InvalidArgument("estimated_mean must be 1-dimensional",
555 estimated_mean.shape().DebugString()));
562 context, estimated_mean.dim_size(0) == 0,
563 errors::InvalidArgument("estimated_mean must be empty for training"
[all...] |
| conv_ops_fused_impl.h | 409 const Tensor& estimated_mean = context->input(4); local
418 TF_REQUIRES(estimated_mean.dims() == 1,
419 errors::InvalidArgument("estimated_mean must be 1-dimensional",
420 estimated_mean.shape().DebugString()));
428 InitDataPtr(estimated_mean, &args->estimated_mean_data);
[all...] |
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_dnn.h | 220 const DeviceMemory<float>& estimated_mean,
233 const DeviceMemory<float>& estimated_mean,
561 const DeviceMemory<U>& estimated_mean,
[all...] |
| cuda_dnn.cc | [all...] |
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| rocm_dnn.h | 211 const DeviceMemory<float>& estimated_mean,
224 const DeviceMemory<float>& estimated_mean,
[all...] |
| rocm_dnn.cc | [all...] |
| /external/tensorflow/tensorflow/stream_executor/ |
| dnn.h | [all...] |
| stream.h | 235 const DeviceMemory<float> &estimated_mean,
256 const DeviceMemory<float> &estimated_mean,
[all...] |
| stream.cc | 342 const DeviceMemory<float> &estimated_mean,
356 this, x, scale, offset, estimated_mean, estimated_variance, x_desc,
392 const DeviceMemory<float> &estimated_mean,
406 this, x, scale, offset, estimated_mean, estimated_variance, x_desc,
[all...] |
