Searched refs:IsSplitV (Results 1 – 7 of 7) sorted by relevance
124 if (IsSplitV(node)) { in GetTransposer()
1700 DCHECK(IsSplitV(*node->node())); in TransposeNode()2060 IsSqueeze(node) || IsSplitV(node) || IsStridedSlice(node) || in IsLayoutAgnosticOp()2140 if (IsSplit(*node_def) || IsSplitV(*node_def)) { in GetDataFanoutPorts()
2152 } else if (IsSplitV(*node)) { in Expand()
1476 } else if ((IsSplit(*node) || IsSplitV(*node)) && in IsSupported()
2128 if (IsSplitV(*node) && node->attr().at("num_split").i() == 1) { in RemoveSplitOrSplitV()
176 bool IsSplitV(const NodeDef& node);
537 bool IsSplitV(const NodeDef& node) { return node.op() == "SplitV"; } in IsSplitV() function