Home
last modified time | relevance | path

Searched refs:grad_fn (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dcustom_gradient.py183 result, grad_fn = f(*args)
196 grad_argspec = tf_inspect.getfullargspec(grad_fn)
219 input_grads, variable_grads = grad_fn(*result_grads, variables=variables)
224 input_grads = grad_fn(*result_grads)
259 result, grad_fn = f(*args, **kwargs)
264 grad_argspec = tf_inspect.getfullargspec(grad_fn)
280 input_grads, variable_grads = grad_fn(*result_grads, variables=variables)
285 input_grads = grad_fn(*result_grads)
Dgradients_util.py388 def _MaybeCompile(scope, op, func, grad_fn): argument
403 return grad_fn() # Exit early
406 return grad_fn() # Exit early
422 return grad_fn()
663 grad_fn = None
673 grad_fn = ops.get_gradient_function(op)
685 grad_fn = func_call.python_grad_func
707 if (grad_fn or is_func_call) and has_out_grads:
713 (not grad_fn and is_func_call) or IsTrainable(op.outputs[i])):
727 if grad_fn:
[all …]
Dgradient_checker_v2_test.py150 def grad_fn(dy): function
154 return y, grad_fn
167 def grad_fn(dy): function
171 return y, grad_fn
Dgradient_checker_v2.py166 grad_fn = _prepare(lambda dy, *xs: grad_fn_unprep(*xs, dy=dy),
170 grad = _to_numpy(grad_fn(dy_data, *xs)[0])
184 grad = _to_numpy(grad_fn(dy_data, *xs)[0])
/external/tensorflow/tensorflow/cc/gradients/
Dgrad_testutil.cc26 ops::GradFunc grad_fn; in CallGradFunction() local
28 op.node()->type_string(), &grad_fn)); in CallGradFunction()
29 TF_RETURN_IF_ERROR(grad_fn(scope, op, grad_inputs, grad_outputs)); in CallGradFunction()
/external/tensorflow/tensorflow/python/kernel_tests/
Daccumulate_n_eager_test.py62 grad_fn = backprop.gradients_function(fn)
63 grad = grad_fn(input_vars[0], input_vars[1], input_vars[2])
Dcholesky_op_test.py373 def _BenchmarkGrad(grad_fn, name, device): argument
382 grad = grad_fn(l, grad_matrix)
Dtensor_array_ops_test.py964 grad_fn = backprop.gradients_function(func)
965 v0_grad, state0_grad, var_grad = grad_fn(v0, state0, var, dy=grad_val)
/external/tensorflow/tensorflow/python/eager/
Dbackprop.py129 grad_fn = ops._gradient_registry.lookup(op_name) # pylint: disable=protected-access
130 if grad_fn is None:
133 return grad_fn(mock_op, *out_grads)
195 def grad_fn(*args, **kwds): function
218 return grad_fn
263 def grad_fn(*args, **kwds): function
267 return grad_fn
Dfunction_gradients_test.py253 grad_fn = backprop.implicit_grad(sum_gather)
254 gradient = grad_fn()
Dbackprop_test.py163 grad_fn = backprop.gradients_function(f)
164 self.assertAllEqual(2., grad_fn(1., dy=2.)[0])
/external/tensorflow/tensorflow/cc/framework/
Dgradients.cc374 ops::GradFunc grad_fn; in IsPrimitiveOpWithNoGrad() local
375 Status s = registry_->Lookup(opname, &grad_fn); in IsPrimitiveOpWithNoGrad()
376 return s.ok() && (grad_fn == nullptr); in IsPrimitiveOpWithNoGrad()
383 ops::GradFunc grad_fn; in CallGradFunction() local
384 TF_RETURN_IF_ERROR(registry_->Lookup(op.node()->type_string(), &grad_fn)); in CallGradFunction()
385 TF_RETURN_IF_ERROR(grad_fn(scope_, op, grad_inputs, grad_outputs)); in CallGradFunction()
/external/tensorflow/tensorflow/contrib/eager/python/
Dtfe_test.py74 def grad_fn(_): function
77 return y, grad_fn
/external/tensorflow/tensorflow/contrib/distribute/python/
Dstrategy_test_lib.py124 grad_fn = backprop.implicit_grad(loss)
125 grad_fn = optimizer.get_filtered_grad_fn(grad_fn)
135 g_v = d.extended.call_for_each_replica(grad_fn, args=(one,))
178 grad_fn = backprop.implicit_grad(loss)
188 g_v = d.extended.call_for_each_replica(grad_fn, args=(one,))
Dcollective_all_reduce_strategy_test.py167 def grad_fn(x): function
184 g_v = d.extended.call_for_each_replica(grad_fn, args=[one])
Dparameter_server_strategy_test.py505 def grad_fn(x): function
522 g_v = d.extended.call_for_each_replica(grad_fn, args=(one,))
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Drev_block_lib.py356 grad_fn = self._make_efficient_grad_fn(args, out)
357 return out, grad_fn
628 def grad_fn(*output_grads, **kwargs): function
631 def grad_fn(*output_grads): function
634 return outputs, grad_fn
/external/tensorflow/tensorflow/compiler/tests/
Deager_test.py163 grad_fn = backprop.gradients_function(f)
164 self.assertAllEqual(2., grad_fn(1., dy=2.)[0])
/external/tensorflow/tensorflow/python/training/
Doptimizer.py47 def get_filtered_grad_fn(grad_fn): argument
58 return [(g, v) for g, v in grad_fn(*args, **kwargs) if g is not None]
/external/tensorflow/tensorflow/contrib/eager/python/examples/workshop/
D2_models.ipynb982 " 400 0.002 0.000 2.246 0.006 backprop.py:145(grad_fn)\n",