Home
last modified time | relevance | path

Searched refs:IsReshape (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/
Dop_types.h135 bool IsReshape(const NodeDef& node);
Dop_types.cc411 bool IsReshape(const NodeDef& node) { return (node.op() == "Reshape"); } in IsReshape() function
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc721 if (!IsReshape(*fanout)) { in MaterializeReductionIndices()
1500 if (!IsReshape(node)) { in IsSimplifiableReshape()
Darithmetic_optimizer.cc1853 return IsReshape(*node); in IsSupported()
1861 if (IsReshape(*input) && !HasControlInputs(*input)) { in TrySimplify()