A.- A._ est atteint d'un syndrome de Norrie, affection congénitale qui provoque une cécité bilatérale ainsi que parfois d'autres symptômes tel que retard mental et diabète. L'Office de l'assurance-invalidité pour le canton de Vaud (l'office AI) a pris en charge les mesures médicales pour traiter l'affection oculaire, le diabète insipide, ainsi que l'atteinte motrice. Se fondant sur le chiffre 395 OIC, il a, par décision du 15 août 2000, refusé toute mesure médicale à raison de l'atteinte motrice au-delà du 30 novembre 1998. B.- Le recours déposé par A._ contre cette décision a été rejeté par le Tribunal des assurances du canton de Vaud, par jugement du 20 septembre 2001. C.- A._ interjette recours de droit administratif contre ce jugement. Sous suite de dépens, il demande l'annulation du jugement cantonal et de la décision de l'office AI, le dossier étant renvoyé à l'office AI pour que ce dernier mette en oeuvre une procédure de reconnaissance selon l'art. 1er al. 2 OIC et rende une nouvelle décision à l'issue de cette procédure. L'office AI a conclu au rejet du recours de même que l'Office fédéral des assurances sociales dans ses observations.