Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/
Dop_types.h151 bool IsSoftmax(const NodeDef& node);
Dop_types.cc450 bool IsSoftmax(const NodeDef& node) { return node.op() == "Softmax"; } in IsSoftmax() function
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc1824 if (IsSoftmax(*x) && !IsInPreserveSet(*x) && in TrySimplify()