Update deprecated gradient_fn
argument to diff_method
#1264
+6
−6
gradient_fn
argument to diff_method
#1264