auto begin_type = mlir::dyn_cast<RankedTensorType>(begin_indices.getType());
    auto size_type = mlir::dyn_cast<RankedTensorType>(sizes.getType());

    if (!input_ty || !begin_type || !size_type ||
        !begin_type.hasStaticShape() || !size_type.hasStaticShape() ||
        begin_type.getRank() != 1 || size_type.getRank() != 1) {
      return failure();
    }