Searched refs:IsSoftmax (Results 1 – 3 of 3) sorted by relevance
151 bool IsSoftmax(const NodeDef& node);
450 bool IsSoftmax(const NodeDef& node) { return node.op() == "Softmax"; } in IsSoftmax() function
1824 if (IsSoftmax(*x) && !IsInPreserveSet(*x) && in TrySimplify()