Searched refs:_compute_gradient (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradient_checker.py | 210 def _compute_gradient(x, function
262 ret = [_compute_gradient(xi, x_shapei, dxi, y, y_shape, dyi, x_init_valuei,
326 ret = _compute_gradient(x, x_shape, dx, y, y_shape, dy, x_init_value, delta,
|
| D | gradient_checker_v2.py | 259 def _compute_gradient(f, function
292 return zip(*[_compute_gradient(f, y.shape, dtypes.as_dtype(y.dtype),
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | load_test.py | 792 def _compute_gradient(function): function
799 self.assertEqual(2., _compute_gradient(root.f).numpy())
802 self.assertEqual(2., _compute_gradient(imported.f).numpy())
|