Searched refs:sum_gather (Results 1 – 3 of 3) sorted by relevance
426 def sum_gather(): function429 grad_fn = backprop.implicit_grad(sum_gather)431 defun_grad_fn = backprop.implicit_grad(def_function.function(sum_gather))
1464 def sum_gather(): function1467 defined = def_function.function(sum_gather)1468 self.assertAllEqual(sum_gather(), defined())1581 def sum_gather(): function1586 defined = function.defun(sum_gather)1589 expected = self.evaluate(sum_gather())
1142 def sum_gather(): function1148 defined = function.Defun()(sum_gather)1155 expected = self.evaluate(sum_gather())