Searched defs:IsArgMax (Results 1 – 2 of 2) sorted by relevance
96 bool IsArgMax(const NodeDef& node) { return node.op() == "ArgMax"; } in IsArgMax() function
108 bool TFGraphDialect::IsArgMax(TFOp op) const { in IsArgMax() function in mlir::tfg::TFGraphDialect