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