Home
last modified time | relevance | path

Searched refs:IsArgMax (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/
Dop_types.h40 bool IsArgMax(const NodeDef& node);
Dop_types.cc96 bool IsArgMax(const NodeDef& node) { return node.op() == "ArgMax"; } in IsArgMax() function
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc2874 IsArgMax(*node) || IsArgMin(*node); in IsSupported()
2932 if (IsArgMax(*reduction_node) || IsArgMin(*reduction_node)) { in TrySimplify()
2962 if (IsAnyMax(node) || IsArgMax(node)) { in FlipMinMax()