Home
last modified time | relevance | path

Searched defs:grad_inputs (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/cc/gradients/
Darray_grad.cc42 const std::vector<Output>& grad_inputs, in PackGrad()
59 const std::vector<Output>& grad_inputs, in UnpackGrad()
69 const std::vector<Output>& grad_inputs, in IdentityGrad()
77 const std::vector<Output>& grad_inputs, in RefIdentityGrad()
85 const std::vector<Output>& grad_inputs, in QuantizeAndDequantizeGrad()
94 const std::vector<Output>& grad_inputs, in QuantizeAndDequantizeV4GradHelper()
113 const std::vector<Output>& grad_inputs, in QuantizeAndDequantizeV3Grad()
124 const std::vector<Output>& grad_inputs, in SplitGrad()
133 const std::vector<Output>& grad_inputs, in SplitVGrad()
147 const std::vector<Output>& grad_inputs, in FillGrad()
[all …]
Dmath_grad.cc58 const std::vector<Output>& grad_inputs, in AbsGrad()
67 const std::vector<Output>& grad_inputs, in NegGrad()
76 const std::vector<Output>& grad_inputs, in InvGrad()
87 const std::vector<Output>& grad_inputs, in SquareGrad()
100 const std::vector<Output>& grad_inputs, in SqrtGrad()
110 const std::vector<Output>& grad_inputs, in RsqrtGrad()
120 const std::vector<Output>& grad_inputs, in ExpGrad()
132 const std::vector<Output>& grad_inputs, in Expm1Grad()
145 const std::vector<Output>& grad_inputs, in LogGrad()
158 const std::vector<Output>& grad_inputs, in Log1pGrad()
[all …]
Dnn_grad.cc28 const std::vector<Output>& grad_inputs, in SoftmaxGrad()
78 const std::vector<Output>& grad_inputs, in SoftmaxCrossEntropyWithLogitsGrad()
117 const std::vector<Output>& grad_inputs, in LogSoftmaxGrad()
129 const std::vector<Output>& grad_inputs, in ReluGradHelper()
138 const std::vector<Output>& grad_inputs, in Relu6GradHelper()
147 const std::vector<Output>& grad_inputs, in LeakyReluGradHelper()
160 const std::vector<Output>& grad_inputs, in LeakyReluGradGradHelper()
174 const std::vector<Output>& grad_inputs, in EluGradHelper()
183 const std::vector<Output>& grad_inputs, in SeluGradHelper()
192 const std::vector<Output>& grad_inputs, in L2LossGrad()
[all …]
Dimage_grad.cc33 const std::vector<Output>& grad_inputs, in ResizeNearestNeighborGradHelper()
55 const std::vector<Output>& grad_inputs, in ResizeBilinearGradHelper()
73 const std::vector<Output>& grad_inputs, in ResizeBicubicGradHelper()
92 const std::vector<Output>& grad_inputs, in ScaleAndTranslateGradHelper()
113 const std::vector<Output>& grad_inputs, in CropAndResizeGradHelper()
Ddata_flow_grad.cc45 const std::vector<Output>& grad_inputs, in DynamicPartitionGrad()
110 const std::vector<Output>& grad_inputs, in DynamicStitchGrad()
Dresource_variable_grad.cc27 const std::vector<Output>& grad_inputs, in ReadVariableOpGrad()
Dgrad_testutil.cc24 const std::vector<Output>& grad_inputs, in CallGradFunction()
Dmanip_grad.cc26 const std::vector<Output>& grad_inputs, in RollGrad()
Dfunctional_grad.cc28 const std::vector<Output>& grad_inputs, in PartitionedCallGrad()
Dlinalg_grad.cc360 const std::vector<Output>& grad_inputs, in EinsumGrad()
/external/tensorflow/tensorflow/c/experimental/gradients/
Dmath_grad.cc55 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
76 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
100 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
126 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
222 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
241 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
278 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
325 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
396 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
Dnn_grad.cc49 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
101 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
125 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
Dnot_differentiable.cc21 absl::Span<AbstractTensorHandle*> grad_inputs) { in Compute()
Darray_grad.cc26 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
Dcustom_gradient_test.cc49 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
/external/tensorflow/tensorflow/cc/framework/
Dgradients.cc144 const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) in SymbolicGradientBuilder()
382 const std::vector<Output>& grad_inputs, in CallGradFunction()
561 const std::vector<Output>& grad_inputs, in AddSymbolicGradients()
572 std::vector<Output> grad_inputs; in AddSymbolicGradients() local
Dwhile_gradients.cc137 const std::vector<Output>& grad_inputs, in AddWhileGradientLoop()
180 const std::vector<Output>& grad_inputs, in AddWhileLoopGradient()
/external/pytorch/torch/csrc/autograd/functions/
Dtensor.cpp22 variable_list grad_inputs(2); in apply() local
154 variable_list grad_inputs(num_outputs()); in apply_impl() local
/external/pytorch/torch/csrc/lazy/ts_backend/
Dts_autograd_functions.cpp55 torch::autograd::variable_list grad_inputs = { in backward() local
/external/pytorch/torch/distributed/pipelining/
D_backward.py186 def hook(grad_inputs): argument
/external/pytorch/torch/csrc/autograd/
Dautograd.cpp140 variable_list grad_inputs = Engine::get_default_engine().execute( in run_backward() local
Dpython_hook.cpp167 THPObjectPtr grad_inputs(wrap_variables(_outputs)); in operator ()() local
Dpython_function.cpp635 variable_list grad_inputs) { in _wrap_outputs()
/external/pytorch/torch/csrc/jit/runtime/
Dautodiff.cpp167 auto grad_inputs = insertGraph(*graph, *bw_graph, grad); in build_script_grad() local
433 value_list grad_inputs = in addReverseInline() local
/external/pytorch/test/
Dtest_foreach.py1445 def hook(grad_inputs, grad_outputs) -> None: argument

12