| /aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/ |
| H A D | math_grad.cc | 59 std::vector<Output>* grad_outputs) { in AbsGrad()
68 std::vector<Output>* grad_outputs) { in NegGrad()
77 std::vector<Output>* grad_outputs) { in InvGrad()
88 std::vector<Output>* grad_outputs) { in SquareGrad()
101 std::vector<Output>* grad_outputs) { in SqrtGrad()
111 std::vector<Output>* grad_outputs) { in RsqrtGrad()
121 std::vector<Output>* grad_outputs) { in ExpGrad()
133 std::vector<Output>* grad_outputs) { in Expm1Grad()
146 std::vector<Output>* grad_outputs) { in LogGrad()
159 std::vector<Output>* grad_outputs) { in Log1pGrad()
[all …]
|
| H A D | array_grad.cc | 43 std::vector<Output>* grad_outputs) { in PackGrad()
60 std::vector<Output>* grad_outputs) { in UnpackGrad()
70 std::vector<Output>* grad_outputs) { in IdentityGrad()
78 std::vector<Output>* grad_outputs) { in RefIdentityGrad()
86 std::vector<Output>* grad_outputs) { in QuantizeAndDequantizeGrad()
95 std::vector<Output>* grad_outputs) { in QuantizeAndDequantizeV4GradHelper()
114 std::vector<Output>* grad_outputs) { in QuantizeAndDequantizeV3Grad()
125 std::vector<Output>* grad_outputs) { in SplitGrad()
134 std::vector<Output>* grad_outputs) { in SplitVGrad()
148 std::vector<Output>* grad_outputs) { in FillGrad()
[all …]
|
| H A D | nn_grad.cc | 29 std::vector<Output>* grad_outputs) { in SoftmaxGrad()
79 std::vector<Output>* grad_outputs) { in SoftmaxCrossEntropyWithLogitsGrad()
118 std::vector<Output>* grad_outputs) { in LogSoftmaxGrad()
130 std::vector<Output>* grad_outputs) { in ReluGradHelper()
139 std::vector<Output>* grad_outputs) { in Relu6GradHelper()
148 std::vector<Output>* grad_outputs) { in LeakyReluGradHelper()
161 std::vector<Output>* grad_outputs) { in LeakyReluGradGradHelper()
175 std::vector<Output>* grad_outputs) { in EluGradHelper()
184 std::vector<Output>* grad_outputs) { in SeluGradHelper()
193 std::vector<Output>* grad_outputs) { in L2LossGrad()
[all …]
|
| H A D | image_grad.cc | 34 std::vector<Output>* grad_outputs) { in ResizeNearestNeighborGradHelper()
56 std::vector<Output>* grad_outputs) { in ResizeBilinearGradHelper()
74 std::vector<Output>* grad_outputs) { in ResizeBicubicGradHelper()
93 std::vector<Output>* grad_outputs) { in ScaleAndTranslateGradHelper()
114 std::vector<Output>* grad_outputs) { in CropAndResizeGradHelper()
|
| H A D | data_flow_grad.cc | 46 std::vector<Output>* grad_outputs) { in DynamicPartitionGrad()
111 std::vector<Output>* grad_outputs) { in DynamicStitchGrad()
|
| H A D | resource_variable_grad.cc | 28 std::vector<Output>* grad_outputs) { in ReadVariableOpGrad()
|
| H A D | grad_testutil.cc | 25 std::vector<Output>* grad_outputs) { in CallGradFunction()
|
| H A D | manip_grad.cc | 27 std::vector<Output>* grad_outputs) { in RollGrad()
|
| H A D | functional_grad.cc | 29 std::vector<Output>* grad_outputs) { in PartitionedCallGrad()
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/framework/ |
| H A D | gradients_test.cc | 105 std::vector<Output> grad_outputs; in TEST_F() local
132 std::vector<Output> grad_outputs; in TEST_F() local
164 std::vector<Output> grad_outputs; in TEST_F() local
201 std::vector<Output> grad_outputs; in TEST_F() local
232 std::vector<Output> grad_outputs; in TEST_F() local
266 std::vector<Output> grad_outputs; in TEST_F() local
298 std::vector<Output> grad_outputs; in TEST_F() local
329 std::vector<Output> grad_outputs; in TEST_F() local
358 std::vector<Output> grad_outputs; in TEST_F() local
395 std::vector<Output> grad_outputs; in TEST_F() local
[all …]
|
| H A D | gradients.cc | 144 const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) in SymbolicGradientBuilder()
383 std::vector<Output>* grad_outputs) { in CallGradFunction()
562 std::vector<Output>* grad_outputs) { in AddSymbolicGradients()
571 std::vector<Output>* grad_outputs) { in AddSymbolicGradients()
|
| H A D | while_gradients.cc | 140 std::vector<Output>* grad_outputs) { in AddWhileGradientLoop()
181 std::vector<Output>* grad_outputs) { in AddWhileLoopGradient()
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/experimental/gradients/ |
| H A D | math_grad.cc | 54 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
75 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
99 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
125 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
221 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
240 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
277 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
324 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
395 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
|
| H A D | nn_grad.cc | 48 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
100 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
124 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
|
| H A D | not_differentiable.cc | 20 AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
|
| H A D | array_grad.cc | 25 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
|
| H A D | custom_gradient_test.cc | 48 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | autograd.cpp | 28 const variable_list& grad_outputs) { in _make_grads()
92 const variable_list& grad_outputs, in run_backward()
187 const variable_list& grad_outputs, in grad()
|
| H A D | python_hook.cpp | 143 THPObjectPtr grad_outputs(wrap_variables(grad_outputs_)); in operator ()() local
168 THPObjectPtr grad_outputs(wrap_variables(_inputs)); in operator ()() local
|
| /aosp_15_r20/external/pytorch/torch/distributed/nn/ |
| H A D | functional.py | 267 def backward(ctx, *grad_outputs):
333 def backward(ctx, *grad_outputs):
399 def backward(ctx, *grad_outputs):
|
| /aosp_15_r20/external/pytorch/torch/nn/parallel/ |
| H A D | _functions.py | 32 def backward(ctx, *grad_outputs):
49 def backward(ctx, *grad_outputs):
|
| /aosp_15_r20/external/pytorch/test/cpp_extensions/ |
| H A D | identity.cpp | 12 static tensor_list backward(AutogradContext* ctx, tensor_list grad_outputs) { in backward()
|
| /aosp_15_r20/external/pytorch/torch/utils/ |
| H A D | __init__.py | 64 def error_pre_hook(grad_outputs): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/ |
| H A D | c_api_test.cc | 1603 TF_Output grad_outputs[2]; in TestGradientsSuccess() local
1627 TF_Output grad_outputs[1]; in TestGradientsError() local
1643 void RunGraphsAndCompareOutputs(TF_Output* grad_outputs, in RunGraphsAndCompareOutputs()
1678 int noutputs, TF_Output* grad_outputs) { in AddGradients()
1838 TF_Output grad_outputs[2]; in BuildGraphAndAddGradientsWithPrefixes() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/ |
| H A D | _functions.cpp | 87 variable_list grad_outputs) { in backward()
|