Searched refs:IsOnesLike (Results 1 – 3 of 3) sorted by relevance
130 bool IsOnesLike(const NodeDef& node);
407 bool IsOnesLike(const NodeDef& node) { return node.op() == "OnesLike"; } in IsOnesLike() function
887 (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()