AlvaroMaza

Changed it so that it checks if the backend is tensorflow

> @AlvaroMaza Hello! I have a pr right now. And I am changing this file too. Would you be comfortable if I added you as a Coauthor (Co-authored-by) to acknowledge...