Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/
Dop_types.h207 bool IsZerosLike(const NodeDef& node);
Dop_types.cc657 bool IsZerosLike(const NodeDef& node) { return node.op() == "ZerosLike"; } in IsZerosLike() function
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc887 (IsZerosLike(*node) ? 0.0 : (IsOnesLike(*node) ? 1.0 : -1.0)); in MaterializeConstantValuedNode()
934 } else if (IsFill(node) || IsZerosLike(node) || IsOnesLike(node)) { in MaterializeConstants()
996 (IsFill(node) || IsZerosLike(node) || IsOnesLike(node))) in IsFoldableUncached()
1738 if (IsZerosLike(node)) return false; in IsOnes()
1777 if (IsZerosLike(node)) return true; in IsZeros()