Searched refs:IsShapeN (Results 1 – 4 of 4) sorted by relevance
147 bool IsShapeN(const NodeDef& node);
438 bool IsShapeN(const NodeDef& node) { return node.op() == "ShapeN"; } in IsShapeN() function
361 if (IsShapeN(node) || IsIdentityN(node) || IsAddN(node) || IsMerge(node)) { in DataInputPos()2094 } else if (IsShape(*node) || IsShapeN(*node)) { in Expand()
1420 } else if (IsShapeN(node)) { in MaybeUpdateNodeContextOutput()