Function IsShapeConsumer

tensorflow/core/grappler/utils.cc:68–71

Is 'node' an operator that consumes only the shape of its input, not the data itself? TODO(ezhulenev): move to op_types.h. Requires to break circular dependency. TODO(ezhulenev): what about Identity passing tensor to Shape consumer?

// TODO(ezhulenev): move to op_types.h. Requires to break circular dependency.
// TODO(ezhulenev): what about Identity passing tensor to Shape consumer?
bool IsShapeConsumer(const NodeDef& node) {
  const string& op = node.op();
  return op == "Shape" || op == "ShapeN" || op == "Rank" || op == "Size";
}

}  // namespace