Searched defs:output_gradients (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scaling_gradient_tape.py | 130 output_gradients=None, argument
206 output_gradients, unconnected_gradients): argument
273 def replica_fn(gradient_tape, target, flattened_sources, output_gradients, argument
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | imperative_grad.py | 36 output_gradients=None, argument
|
| D | backprop.py | 990 output_gradients=None, argument
|
| D | pywrap_tfe_src.cc | 1333 tensorflow::gtl::ArraySlice<PyObject*> output_gradients, in CallBackwardFunction()
2719 PyObject* sources, PyObject* output_gradients,
|
| /external/tensorflow/tensorflow/python/framework/experimental/ |
| D | tape.py | 41 def gradient(self, targets, sources, output_gradients=None): argument
|
| D | tape.cc | 55 std::vector<AbstractTensorHandle*> output_gradients) { in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | pooling_ops_3d_test.py | 34 def _AvgPoolGrad(inputs, outputs, output_gradients, ksize, strides, padding): argument
|
| D | pooling_ops_test.py | 554 def AvgPoolGrad(inputs, outputs, output_gradients, ksize, strides, padding, argument
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | tape.h | 601 gtl::ArraySlice<Gradient*> output_gradients, const TensorTape& tensor_tape, in InitialGradients()
687 gtl::ArraySlice<Gradient*> output_gradients, absl::Span<Gradient*> result, in ComputeGradient()
|
| D | gradients.cc | 177 gtl::ArraySlice<AbstractTensorHandle*> output_gradients, in CallBackwardFunction()
274 absl::Span<AbstractTensorHandle* const> output_gradients, in ComputeGradient()
|
| /external/tensorflow/tensorflow/python/ |
| D | tfe_wrapper.cc | 933 const py::handle& unconnected_gradients) { in PYBIND11_MODULE()
|