Searched refs:IsReshape (Results 1 – 4 of 4) sorted by relevance
155 bool IsReshape(const NodeDef& node);
484 bool IsReshape(const NodeDef& node) { return (node.op() == "Reshape"); } in IsReshape() function
1952 return (IsReshape(*node) || IsBroadcastTo(*node)) && in IsSupported()1961 if (IsReshape(*node) && IsReshape(*input) && !IsInPreserveSet(*input)) { in TrySimplify()
776 if (!IsReshape(*fanout)) { in MaterializeReductionIndices()1670 if (!IsReshape(node)) { in IsSimplifiableReshape()