Home
last modified time | relevance | path

Searched refs:output_gradients (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/python/training/experimental/
Dloss_scaling_gradient_tape.py130 output_gradients=None, argument
176 args=(self, self._loss_scale, target, sources, output_gradients,
206 output_gradients, unconnected_gradients): argument
273 def replica_fn(gradient_tape, target, flattened_sources, output_gradients, argument
281 scaled_target, flattened_sources, output_gradients,
308 output_gradients, initial_grads))
Dloss_scaling_gradient_tape_test.py116 return g.gradient(y, x, output_gradients=constant_op.constant(2.0))
/external/tensorflow/tensorflow/compiler/tests/
Dpooling_ops_test.py349 output_gradients = array_ops.placeholder(
354 output_gradients,
362 output_gradients: output_gradient_vals})
385 xla_output_gradients = output_gradients
392 xla_output_gradients = NHWCToNCHW(output_gradients)
420 output_gradients: output_gradient_vals
554 def AvgPoolGrad(inputs, outputs, output_gradients, ksize, strides, padding, argument
559 output_gradients,
Dpooling_ops_3d_test.py34 def _AvgPoolGrad(inputs, outputs, output_gradients, ksize, strides, padding): argument
38 output_gradients,
234 output_gradients = array_ops.placeholder(
239 output_gradients,
246 output_gradients: output_gradient_vals})
270 output_gradients,
287 output_gradients: output_gradient_vals
/external/tensorflow/tensorflow/python/eager/
Dimperative_grad.py36 output_gradients=None, argument
75 output_gradients,
Dbackprop.py579 this_tape, nest.flatten(result), sources, output_gradients=dy)
990 output_gradients=None, argument
1070 if output_gradients is not None:
1071 output_gradients = [None if x is None else ops.convert_to_tensor(x)
1072 for x in nest.flatten(output_gradients)]
1078 output_gradients=output_gradients,
Dpywrap_tfe.h254 PyObject* sources, PyObject* output_gradients,
Dforwardprop.py138 grads, forwardprop_aids, output_gradients=nontrivial_tangents)
Dpywrap_tfe_src.cc1333 tensorflow::gtl::ArraySlice<PyObject*> output_gradients, in CallBackwardFunction() argument
1335 PyObject* grads = PyTuple_New(output_gradients.size()); in CallBackwardFunction()
1336 for (int i = 0; i < output_gradients.size(); ++i) { in CallBackwardFunction()
1337 if (output_gradients[i] == nullptr) { in CallBackwardFunction()
1342 reinterpret_cast<PyObject*>(output_gradients[i])); in CallBackwardFunction()
2719 PyObject* sources, PyObject* output_gradients, argument
2771 if (output_gradients != Py_None) {
2772 outgrad_vec = MakeTensorList(output_gradients);
/external/tensorflow/tensorflow/python/framework/experimental/
Dtape.py41 def gradient(self, targets, sources, output_gradients=None): argument
46 output_gradients or [])
Dtape.cc55 std::vector<AbstractTensorHandle*> output_gradients) { in PYBIND11_MODULE() argument
58 source_tensors, output_gradients, in PYBIND11_MODULE()
/external/tensorflow/tensorflow/c/eager/
Dgradients.cc113 gtl::ArraySlice<AbstractTensorHandle*> output_gradients,
177 gtl::ArraySlice<AbstractTensorHandle*> output_gradients, in CallBackwardFunction() argument
186 return gradient_function->Compute(ctx_, output_gradients, result); in CallBackwardFunction()
274 absl::Span<AbstractTensorHandle* const> output_gradients, in ComputeGradient() argument
293 output_gradients, result, /*build_default_zeros_grads*/ false)); in ComputeGradient()
Dtape.h102 gtl::ArraySlice<Gradient*> output_gradients,
178 gtl::ArraySlice<Gradient*> output_gradients, absl::Span<Gradient*> result,
601 gtl::ArraySlice<Gradient*> output_gradients, const TensorTape& tensor_tape, in InitialGradients() argument
606 if (output_gradients.empty() || output_gradients[i] == nullptr) { in InitialGradients()
644 (*result)[id].push_back(output_gradients[i]); in InitialGradients()
687 gtl::ArraySlice<Gradient*> output_gradients, absl::Span<Gradient*> result, in ComputeGradient() argument
697 sources_that_are_targets, output_gradients, in ComputeGradient()
Dgradients.h169 absl::Span<AbstractTensorHandle* const> output_gradients,
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.-gradient-tape.pbtxt15 …argspec: "args=[\'self\', \'target\', \'sources\', \'output_gradients\', \'unconnected_gradients\'…
Dtensorflow.autodiff.-gradient-tape.pbtxt15 …argspec: "args=[\'self\', \'target\', \'sources\', \'output_gradients\', \'unconnected_gradients\'…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-gradient-tape.pbtxt15 …argspec: "args=[\'self\', \'target\', \'sources\', \'output_gradients\', \'unconnected_gradients\'…
/external/tensorflow/tensorflow/python/eager/benchmarks/resnet50/
Dhvp_test.py75 grads, model.trainable_variables, output_gradients=vector)
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dmath_test.py64 for output_gradients in (None, out * math_ops.cast(i, out.dtype)):
65 grad = g.gradient(out, x_i, output_gradients=output_gradients)
Dcontrol_flow_ops_test.py512 grad = g.gradient(loss, x1, output_gradients=ones)
533 grad = g.gradient(loss, x1, output_gradients=ones)
554 grad = g.gradient(loss, x1, output_gradients=ones)
/external/tensorflow/tensorflow/python/ops/
Dcontrol_flow_ops_test.py444 cond_true, [true_in, false_in], output_gradients=3.)
446 cond_false, [true_in, false_in], output_gradients=3.)
1091 actual_grads = tape.gradient(case_out, inputs, output_gradients=out_grad)
1127 actual_grads = tape.gradient(case_out, inputs, output_gradients=out_grad)
1137 tape.gradient(y, inputs[used_bi], output_gradients=out_grad)))
Dcustom_gradient.py540 output_gradients=dresult,
Dscript_ops.py368 return tape.gradient(eager_outputs, eager_inputs, output_gradients=dy)
/external/tensorflow/tensorflow/python/keras/integration_test/
Dforwardprop_test.py263 first_grads, model.trainable_variables, output_gradients=vector)
/external/tensorflow/tensorflow/python/
Dtfe_wrapper.cc931 const py::handle& sources, const py::handle& output_gradients, in PYBIND11_MODULE()
937 tape.ptr(), target.ptr(), sources.ptr(), output_gradients.ptr(), in PYBIND11_MODULE()

12