Searched refs:IsArgMax (Results 1 – 3 of 3) sorted by relevance
35 bool IsArgMax(const NodeDef& node);
70 bool IsArgMax(const NodeDef& node) { return node.op() == "ArgMax"; } in IsArgMax() function
2784 IsArgMax(*node) || IsArgMin(*node); in IsSupported()2822 if (IsArgMax(*reduction_node) || IsArgMin(*reduction_node)) { in TrySimplify()2851 if (IsAnyMax(node) || IsArgMax(node)) { in FlipMinMax()