Searched refs:IsRank (Results 1 – 3 of 3) sorted by relevance
125 bool IsRank(const NodeDef& node);
381 bool IsRank(const NodeDef& node) { return node.op() == "Rank"; } in IsRank() function
1386 } else if (IsRank(node)) { in MaybeUpdateNodeContextOutput()