Searched refs:sum_gather (Results 1 – 3 of 3) sorted by relevance
250 def sum_gather(): function253 grad_fn = backprop.implicit_grad(sum_gather)255 defun_grad_fn = backprop.implicit_grad(def_function.function(sum_gather))
888 def sum_gather(): function891 defined = def_function.function(sum_gather)892 self.assertAllEqual(sum_gather(), defined())970 def sum_gather(): function975 defined = function.defun(sum_gather)978 expected = self.evaluate(sum_gather())
1085 def sum_gather(): function1091 defined = function.Defun()(sum_gather)1098 expected = self.evaluate(sum_gather())