HomeSort by relevance Sort by last modified time
    Searched defs:variance_epsilon (Results 1 - 3 of 3) sorted by null

  /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);
187 T variance_epsilon, bool scale_after_normalization, \
243 typename TTypes<T, 4>::ConstTensor out_backprop, T variance_epsilon, \
    [all...]
  /external/tensorflow/tensorflow/tools/graph_transforms/
fold_old_batch_norms.cc 55 const string epsilon_attr = is_fused ? "epsilon" : "variance_epsilon";
75 const float variance_epsilon = batch_norm_node.attr().at(epsilon_attr).f(); local
90 (1.0f / sqrtf(variance.flat<float>()(i) + variance_epsilon)) *
96 (1.0f / sqrtf(variance.flat<float>()(i) + variance_epsilon));
  /external/tensorflow/tensorflow/core/grappler/optimizers/
remapper.cc 711 NodeDef* variance_epsilon = optimized_graph->add_node(); local
714 variance_epsilon));
715 variance_epsilon->set_device(fused_node.device());
724 *variance_plus_epsilon->add_input() = variance_epsilon->name();
    [all...]

Completed in 192 milliseconds