Home
last modified time | relevance | path

Searched refs:IsOnesLike (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/
Dop_types.h130 bool IsOnesLike(const NodeDef& node);
Dop_types.cc407 bool IsOnesLike(const NodeDef& node) { return node.op() == "OnesLike"; } in IsOnesLike() function
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc887 (IsZerosLike(*node) ? 0.0 : (IsOnesLike(*node) ? 1.0 : -1.0)); in MaterializeConstantValuedNode()
934 } else if (IsFill(node) || IsZerosLike(node) || IsOnesLike(node)) { in MaterializeConstants()
996 (IsFill(node) || IsZerosLike(node) || IsOnesLike(node))) in IsFoldableUncached()
1737 if (IsOnesLike(node)) return true; in IsOnes()
1776 if (IsOnesLike(node)) return false; in IsZeros()