Searched refs:IsArgMax (Results 1 – 3 of 3) sorted by relevance
40 bool IsArgMax(const NodeDef& node);
96 bool IsArgMax(const NodeDef& node) { return node.op() == "ArgMax"; } in IsArgMax() function
2874 IsArgMax(*node) || IsArgMin(*node); in IsSupported()2932 if (IsArgMax(*reduction_node) || IsArgMin(*reduction_node)) { in TrySimplify()2962 if (IsAnyMax(node) || IsArgMax(node)) { in FlipMinMax()