Searched refs:old_mean (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | fused_batchnorm_test.py | 47 def _reference_training(self, x, scale, offset, old_mean, old_var, epsilon, argument
62 exponential_avg_factor) * old_mean + exponential_avg_factor * mean
163 old_mean = None
166 old_mean = array_ops.placeholder(
174 mean=old_mean,
191 old_mean: mean_val,
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | robust_stats.cc | 145 double old_mean = mean_; in HuberMAD() local
147 if (mean_ == old_mean) break; in HuberMAD()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batch_norm_op.cc | 138 xla::XlaOp old_mean = ctx->Input(3); in CompileImpl() local
140 xla::ScalarLike(old_mean, 1.0f - exponential_avg_factor_); in CompileImpl()
141 xla::XlaOp beta = xla::ScalarLike(old_mean, exponential_avg_factor_); in CompileImpl()
144 xla::Add(xla::Mul(old_mean, alpha), in CompileImpl()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_fused_batchnorm_test.py | 101 def _running_mean(self, old_mean, new_val, factor): argument
105 return (1.0 - factor) * old_mean + factor * new_val
107 def _training_ref(self, x, scale, offset, old_mean, old_var, argument
136 mean = self._running_mean(old_mean, batch_mean, exponential_avg_factor)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 148 typename TTypes<U>::ConstVec old_mean(running_mean_input.vec<U>()); in operator ()() local
214 one_minus_factor * old_mean + exponential_avg_factor * batch_mean; in operator ()()
|