Searched refs:IsAddN (Results 1 – 9 of 9) sorted by relevance
112 bool IsAddN(const tensorflow::NodeDef& node_def, int n);
358 bool IsAddN(const tensorflow::NodeDef& node_def, int n) { in IsAddN() function
555 } else if (IsAddN(n, 2)) { in TEST()
26 bool IsAddN(const NodeDef& node);
37 bool IsAddN(const NodeDef& node) { return node.op() == "AddN"; } in IsAddN() function
69 return IsAddN(node) && NumNonControlInputs(node) <= 1; in IsTrivialOp()
361 if (IsShapeN(node) || IsIdentityN(node) || IsAddN(node) || IsMerge(node)) { in DataInputPos()2065 if (IsAddN(*node)) { in Expand()
515 if (!IsAddN(node) && node.op() != "AccumulateNV2") { in SchedulingPass()
500 if (!IsAdd(node) && !IsAddN(node)) { in CanOptimize()