Searched refs:IsRank (Results 1 – 4 of 4) sorted by relevance
143 bool IsRank(const NodeDef& node);
448 bool IsRank(const NodeDef& node) { return node.op() == "Rank"; } in IsRank() function
396 return !IsShape(*node) && !IsRank(*node); in IsCpuCompatibleDataType()
1681 } else if (IsRank(node)) { in MaybeUpdateNodeContextOutput()