| /external/tensorflow/tensorflow/core/kernels/ |
| batch_norm_op.cc | 39 float variance_epsilon; local
41 context->GetAttr("variance_epsilon", &variance_epsilon));
42 variance_epsilon_ = T(variance_epsilon);
89 float variance_epsilon; local
91 context->GetAttr("variance_epsilon", &variance_epsilon));
92 variance_epsilon_ = T(variance_epsilon);
183 T variance_epsilon, bool scale_after_normalization, \
239 typename TTypes<T, 4>::ConstTensor out_backprop, T variance_epsilon, \
[all...] |
| batch_norm_op.h | 32 typename TTypes<T>::ConstVec gamma, T variance_epsilon,
55 ((var + var.constant(variance_epsilon)).rsqrt() * gamma)
64 ((var + var.constant(variance_epsilon)).rsqrt())
80 T variance_epsilon, bool scale_after_normalization,
117 scratch1.device(d) = (var + var.constant(variance_epsilon)).rsqrt();
143 (var + var.constant(variance_epsilon));
|
| quantized_batch_norm_op.cc | 37 float variance_epsilon, bool scale_after_normalization,
69 sqrtf(var_value + variance_epsilon)) *
74 sqrtf(var_value + variance_epsilon)) +
100 float variance_epsilon, bool scale_after_normalization,
133 scale_value = (1.0f / sqrtf(var_value + variance_epsilon)) * gamma_value;
135 scale_value = (1.0f / sqrtf(var_value + variance_epsilon));
169 context->GetAttr("variance_epsilon", &variance_epsilon_));
|
| quantized_instance_norm.cc | 140 float variance_epsilon, float* minimum, float* maximum) {
143 const float32x4_t eps = vdupq_n_f32(variance_epsilon);
196 const float* variance_ptr, float variance_epsilon,
198 const float32x4_t eps = vdupq_n_f32(variance_epsilon);
259 context->GetAttr("variance_epsilon", &variance_epsilon_));
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| fold_old_batch_norms.cc | 56 const string epsilon_attr = is_fused ? "epsilon" : "variance_epsilon";
76 const float variance_epsilon = batch_norm_node.attr().at(epsilon_attr).f(); local
91 (1.0f / sqrtf(variance.flat<float>()(i) + variance_epsilon)) *
97 (1.0f / sqrtf(variance.flat<float>()(i) + variance_epsilon));
|
| /external/tensorflow/tensorflow/python/ops/ |
| nn_impl.py | 785 variance_epsilon,
[all...] |
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| layers.py | [all...] |
| /external/tensorflow/tensorflow/core/graph/ |
| graph_constructor_test.cc | 1119 key: "variance_epsilon"
[all...] |
| /external/tensorflow/tensorflow/go/op/ |
| wrappers.go | [all...] |
