| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| resample.py | 137 batch_mean = math_ops.reduce_mean(weights)
139 estimated_mean, batch_mean, mean_decay)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| fused_batch_norm_op.cc | 68 typename TTypes<U>::Vec batch_mean(batch_mean_output->vec<U>());
103 batch_mean.device(d) = mean;
225 Tensor* batch_mean, Tensor* batch_var, Tensor* saved_mean,
246 f(context->eigen_device<GPUDevice>(), batch_mean->flat<U>());
303 auto batch_mean_ptr = StreamExecutorUtil::AsDeviceMemory<U>(*batch_mean);
553 Tensor* batch_mean = nullptr; variable
555 context->allocate_output(1, scale.shape(), &batch_mean));
568 y, batch_mean, batch_var, saved_mean, saved_maybe_inv_var,
[all...] |
| mkl_fused_batch_norm_op.cc | 148 Tensor* batch_mean = nullptr; variable
151 AllocateOutputSetMklShape(context, 1, &batch_mean, scale.shape(),
164 mkl_context.MklSetMeanVariance(*batch_mean, *batch_variance);
[all...] |
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| virtual_scheduler_test.cc | 247 auto batch_mean = batch_norm.batch_mean; local
254 batch_mean.op(),
810 auto batch_mean = batch_norm.batch_mean; local
818 s.WithOpName("batch_mean1").WithDevice(kCPU1), batch_mean);
[all...] |
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_dnn.h | 207 DeviceMemory<float>* y, DeviceMemory<float>* batch_mean,
220 DeviceMemory<Eigen::half>* y, DeviceMemory<float>* batch_mean,
[all...] |
| cuda_dnn.cc | [all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| nn_impl.py | [all...] |
| /external/tensorflow/tensorflow/stream_executor/ |
| dnn.h | 906 // batch_mean: batch mean, to be used to compute the running mean.
[all...] |
| stream.h | 228 DeviceMemory<float> *y, DeviceMemory<float> *batch_mean,
249 DeviceMemory<Eigen::half> *y, DeviceMemory<float> *batch_mean,
[all...] |
| stream.cc | 321 DeviceMemory<float> *y, DeviceMemory<float> *batch_mean,
332 scale_offset_desc, epsilon, y, batch_mean, batch_var, saved_mean,
371 DeviceMemory<Eigen::half> *y, DeviceMemory<float> *batch_mean,
382 scale_offset_desc, epsilon, y, batch_mean, batch_var, saved_mean,
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| computation_builder.h | [all...] |
