Searched refs:IsIdentityN (Results 1 – 8 of 8) sorted by relevance
60 if (!IsIdentity(node) && !IsIdentityN(node)) { in SafeToRemoveIdentity()153 IsIdentityN(node) && !IsIdentityNSingleInput(node); in NumEdgesIfBypassed()192 IsIdentityN(node) && !IsIdentityNSingleInput(node); in BypassingNodeIsBeneficial()253 IsIdentityN(*node) && !IsIdentityNSingleInput(*node); in OptimizeNode()471 if (IsNoOp(node) || IsIdentity(node) || IsIdentityN(node) || in OptimizeDependencies()
94 if (IsIdentityN(node)) { in GetTransposer()
236 if (IsIdentityN(*curr_node)) { in IdentityNTerminalPorts()
1389 DCHECK(IsIdentityN(*node->node())); in TransposeNode()2055 IsIdentityN(node) || IsMerge(node) || IsMirrorPad(node) || in IsLayoutAgnosticOp()2122 if (IsShapeN(*node_def) || IsIdentityN(*node_def) || IsAddN(*node_def) || in GetDataFaninPorts()2137 if (IsIdentityN(*node_def) || IsShape(*node_def) || IsShapeN(*node_def)) { in GetDataFanoutPorts()
369 if (IsShapeN(node) || IsIdentityN(node) || IsAddN(node) || IsMerge(node)) { in DataInputPos()2131 } else if (IsIdentityN(*node)) { in Expand()
3561 if (!(IsIdentityN(*node) || IsIdentityNSingleInput(*node)) || in PartialConstPropThroughIdentityN()
96 bool IsIdentityN(const NodeDef& node);
318 bool IsIdentityN(const NodeDef& node) { in IsIdentityN() function324 return IsIdentityN(node) && node.attr().count("T") != 0 && in IsIdentityNSingleInput()