Home
last modified time | relevance | path

Searched refs:IsSplitV (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/
Dop_types.h155 bool IsSplitV(const NodeDef& node);
Dop_types.cc458 bool IsSplitV(const NodeDef& node) { return node.op() == "SplitV"; } in IsSplitV() function
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dlayout_optimizer.cc2098 } else if (IsSplitV(*node)) { in Expand()
Dconstant_folding.cc1927 if (IsSplitV(*node) && node->attr().at("num_split").i() == 1) { in RemoveSplitOrSplitV()
Darithmetic_optimizer.cc1434 } else if ((IsSplit(*node) || IsSplitV(*node)) && in IsSupported()