Searched refs:IsIdentityN (Results 1 – 6 of 6) sorted by relevance
60 if (!IsIdentity(node) && !IsIdentityN(node)) { in SafeToRemoveIdentity()131 IsIdentityN(node) && !IsIdentityNSingleInput(node); in NumEdgesIfBypassed()170 IsIdentityN(node) && !IsIdentityNSingleInput(node); in BypassingNodeIsBeneficial()231 IsIdentityN(*node) && !IsIdentityNSingleInput(*node); in OptimizeNode()448 if (IsNoOp(node) || IsIdentity(node) || IsIdentityN(node) || in OptimizeDependencies()
218 if (IsIdentityN(*curr_node)) { in IdentityNTerminalPorts()
361 if (IsShapeN(node) || IsIdentityN(node) || IsAddN(node) || IsMerge(node)) { in DataInputPos()2077 } else if (IsIdentityN(*node)) { in Expand()
3013 if ((IsIdentityN(*node) || IsIdentityNSingleInput(*node)) && in PartialConstPropThroughIdentityN()
257 bool IsIdentityN(const NodeDef& node) { in IsIdentityN() function263 return IsIdentityN(node) && node.attr().count("T") != 0 && in IsIdentityNSingleInput()
82 bool IsIdentityN(const NodeDef& node);