Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/
Dop_types.h182 bool IsZerosLike(const NodeDef& node);
Dop_types.cc568 bool IsZerosLike(const NodeDef& node) { return node.op() == "ZerosLike"; } in IsZerosLike() function
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc829 (IsZerosLike(*node) ? 0.0 : (IsOnesLike(*node) ? 1.0 : -1.0)); in MaterializeConstantValuedNode()
849 } else if (IsFill(node) || IsZerosLike(node) || IsOnesLike(node)) { in MaterializeConstants()
1568 if (IsZerosLike(node)) return false; in IsOnes()
1607 if (IsZerosLike(node)) return true; in IsZeros()