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