pir::OpResult shape(const pir::Value& input) {
  if (input.type().isa<paddle::dialect::DenseTensorType>()) {
    paddle::dialect::ShapeOp shape_op =
        ApiBuilder::Instance().GetBuilder()->Build<paddle::dialect::ShapeOp>(
            input);
    return shape_op.result(0);
  }
  if (input.type().isa<paddle::dialect::SelectedRowsType>()) {
    paddle::dialect::ShapeSrOp shape_sr_op =
        ApiBuilder::Instance().GetBuilder()->Build<paddle::dialect::ShapeSrOp>(
            input);
    return shape_sr_op.result(0);
  }
  PADDLE_THROW(phi::errors::Unimplemented(
      "The kernel of (shape) for input OpResult is unimplemented, please check "
      "the type of input OpResult."));
}