/external/tensorflow/tensorflow/python/training/experimental/ |
D | loss_scaling_gradient_tape.py | 130 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))
|
D | loss_scaling_gradient_tape_test.py | 116 return g.gradient(y, x, output_gradients=constant_op.constant(2.0))
|
/external/tensorflow/tensorflow/compiler/tests/ |
D | pooling_ops_test.py | 349 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,
|
D | pooling_ops_3d_test.py | 34 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/ |
D | imperative_grad.py | 36 output_gradients=None, argument 75 output_gradients,
|
D | backprop.py | 579 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,
|
D | pywrap_tfe.h | 254 PyObject* sources, PyObject* output_gradients,
|
D | forwardprop.py | 138 grads, forwardprop_aids, output_gradients=nontrivial_tangents)
|
D | pywrap_tfe_src.cc | 1333 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/ |
D | tape.py | 41 def gradient(self, targets, sources, output_gradients=None): argument 46 output_gradients or [])
|
D | tape.cc | 55 std::vector<AbstractTensorHandle*> output_gradients) { in PYBIND11_MODULE() argument 58 source_tensors, output_gradients, in PYBIND11_MODULE()
|
/external/tensorflow/tensorflow/c/eager/ |
D | gradients.cc | 113 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()
|
D | tape.h | 102 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()
|
D | gradients.h | 169 absl::Span<AbstractTensorHandle* const> output_gradients,
|
/external/tensorflow/tensorflow/tools/api/golden/v2/ |
D | tensorflow.-gradient-tape.pbtxt | 15 …argspec: "args=[\'self\', \'target\', \'sources\', \'output_gradients\', \'unconnected_gradients\'…
|
D | tensorflow.autodiff.-gradient-tape.pbtxt | 15 …argspec: "args=[\'self\', \'target\', \'sources\', \'output_gradients\', \'unconnected_gradients\'…
|
/external/tensorflow/tensorflow/tools/api/golden/v1/ |
D | tensorflow.-gradient-tape.pbtxt | 15 …argspec: "args=[\'self\', \'target\', \'sources\', \'output_gradients\', \'unconnected_gradients\'…
|
/external/tensorflow/tensorflow/python/eager/benchmarks/resnet50/ |
D | hvp_test.py | 75 grads, model.trainable_variables, output_gradients=vector)
|
/external/tensorflow/tensorflow/python/ops/parallel_for/ |
D | math_test.py | 64 for output_gradients in (None, out * math_ops.cast(i, out.dtype)): 65 grad = g.gradient(out, x_i, output_gradients=output_gradients)
|
D | control_flow_ops_test.py | 512 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/ |
D | control_flow_ops_test.py | 444 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)))
|
D | custom_gradient.py | 540 output_gradients=dresult,
|
D | script_ops.py | 368 return tape.gradient(eager_outputs, eager_inputs, output_gradients=dy)
|
/external/tensorflow/tensorflow/python/keras/integration_test/ |
D | forwardprop_test.py | 263 first_grads, model.trainable_variables, output_gradients=vector)
|
/external/tensorflow/tensorflow/python/ |
D | tfe_wrapper.cc | 931 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()
|