Searched refs:IsOnesLike (Results 1 – 3 of 3) sorted by relevance
109 bool IsOnesLike(const NodeDef& node);
340 bool IsOnesLike(const NodeDef& node) { return node.op() == "OnesLike"; } in IsOnesLike() function
829 (IsZerosLike(*node) ? 0.0 : (IsOnesLike(*node) ? 1.0 : -1.0)); in MaterializeConstantValuedNode()849 } else if (IsFill(node) || IsZerosLike(node) || IsOnesLike(node)) { in MaterializeConstants()1567 if (IsOnesLike(node)) return true; in IsOnes()1606 if (IsOnesLike(node)) return false; in IsZeros()