Searched refs:IsShapeN (Results 1 – 6 of 6) sorted by relevance
115 if (IsShapeN(node)) { in GetTransposer()
1637 DCHECK(IsShapeN(*node->node())); in TransposeNode()2059 IsShapeN(node) || IsFill(node) || IsSlice(node) || IsSplit(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()2148 } else if (IsShape(*node) || IsShapeN(*node)) { in Expand()
166 bool IsShapeN(const NodeDef& node);
517 bool IsShapeN(const NodeDef& node) { return node.op() == "ShapeN"; } in IsShapeN() function
1715 } else if (IsShapeN(node)) { in MaybeUpdateNodeContextOutput()