Fix doctests that use old gradient function names.