Searched defs:expected_grad (Results 1 – 2 of 2) sorted by relevance
65 def _run_fn_with_grad_check(self, strategy, var, opt, expected_grad): argument
514 const Tensor& expected_grad) { in CheckGrad()596 const Tensor& expected_grad) { in CheckGrad()