Searched refs:output_grads (Results 1 – 2 of 2) sorted by relevance
524 output_grads, argument540 ctrl_dep_grads = [g for g in output_grads if g is not None]543 inputs = _force_data_dependency(output_grads, inputs)564 output_grads)604 def _grad_fn(output_grads, variables=None): argument616 output_grads=output_grads,628 def grad_fn(*output_grads, **kwargs): argument629 return _grad_fn(output_grads, kwargs["variables"])631 def grad_fn(*output_grads): argument632 return _grad_fn(output_grads)
2136 op_outputs](PyObject* output_grads) { argument2142 op_outputs, output_grads));