Home
last modified time | relevance | path

Searched defs:epsilon_attr (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dunfuse_batch_norm.cc66 Value MaterializeEpsilon(Operation* op, FloatAttr epsilon_attr, in MaterializeEpsilon()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_old_batch_norms.cc54 const string epsilon_attr = is_fused ? "epsilon" : "variance_epsilon"; in GetScaleAndOffsetValues() local
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tf.cc1305 auto epsilon_attr = in matchAndRewrite() local
1365 auto epsilon_attr = in matchAndRewrite() local