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