| /external/tensorflow/tensorflow/cc/gradients/ |
| H A D | grad_testutil.cc | 25 std::vector<Output>* grad_outputs) {
29 TF_RETURN_IF_ERROR(grad_fn(scope, op, grad_inputs, grad_outputs));
23 CallGradFunction(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
|
| H A D | data_flow_grad.cc | 46 std::vector<Output>* grad_outputs) {
102 grad_outputs->push_back(Reshape(scope, reconstructed, Shape(scope, data)));
104 grad_outputs->push_back(NoGradient());
111 std::vector<Output>* grad_outputs) {
125 grad_outputs->push_back(NoGradient());
145 grad_outputs->push_back(Gather(scope, grad_inputs[0], index));
44 DynamicPartitionGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
109 DynamicStitchGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
|
| H A D | nn_grad.cc | 29 std::vector<Output>* grad_outputs) {
45 grad_outputs->push_back(dx);
52 std::vector<Output>* grad_outputs) {
57 grad_outputs->push_back(dx);
64 std::vector<Output>* grad_outputs) {
66 grad_outputs->push_back(dx);
73 std::vector<Output>* grad_outputs) {
75 grad_outputs->push_back(dx);
82 std::vector<Output>* grad_outputs) {
84 grad_outputs
27 SoftmaxGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
50 LogSoftmaxGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
62 ReluGradHelper(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
71 Relu6GradHelper(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
80 EluGradHelper(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
89 SeluGradHelper(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
98 L2LossGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
106 BiasAddGradHelper(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
121 Conv2DGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
151 MaxPoolGradHelper(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
175 MaxPoolGradV2Helper(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
199 LRNGradHelper(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
[all...] |
| H A D | array_grad.cc | 44 std::vector<Output>* grad_outputs) {
50 grad_outputs->reserve(N);
53 grad_outputs->emplace_back(o);
61 std::vector<Output>* grad_outputs) {
64 grad_outputs->push_back(Stack(scope, grad_inputs, Stack::Axis(axis)));
71 std::vector<Output>* grad_outputs) {
72 grad_outputs->push_back(Identity(scope, grad_inputs[0]));
79 std::vector<Output>* grad_outputs) {
80 grad_outputs->push_back(Identity(scope, grad_inputs[0]));
87 std::vector<Output>* grad_outputs) {
42 PackGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
59 UnpackGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
69 IdentityGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
77 RefIdentityGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
85 QuantizeAndDequantizeGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
93 QuantizeAndDequantizeV2Grad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
103 QuantizeAndDequantizeV3Grad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
114 SplitGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
123 DiagGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
131 DiagPartGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
139 MatrixDiagGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
147 MatrixBandPartGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
160 GatherNdGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
172 CheckNumericsGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
185 ReshapeGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
195 ExpandDimsGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
205 SqueezeGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
214 TransposeGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
224 ReverseSequenceGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
240 ReverseGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
250 ScatterNdGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
261 ScatterNdNonAliasingAddGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
273 PadGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
294 SpaceToBatchGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
307 SpaceToBatchNDGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
318 BatchToSpaceGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
331 BatchToSpaceNDGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
342 SpaceToDepthGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
353 DepthToSpaceGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
364 MirrorPadGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
377 MirrorPadGradGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
[all...] |
| H A D | math_grad.cc | 57 std::vector<Output>* grad_outputs) {
59 grad_outputs->push_back(Mul(scope, grad_inputs[0], Sign(scope, op.input(0))));
66 std::vector<Output>* grad_outputs) {
68 grad_outputs->push_back(Neg(scope, grad_inputs[0]));
75 std::vector<Output>* grad_outputs) {
77 grad_outputs->push_back(
86 std::vector<Output>* grad_outputs) {
91 grad_outputs->push_back(
99 std::vector<Output>* grad_outputs) {
101 grad_outputs
55 AbsGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
64 NegGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
73 InvGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
84 SquareGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
97 SqrtGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
107 RsqrtGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
117 ExpGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
129 Expm1Grad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
142 LogGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
155 Log1pGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
169 SinhGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
182 CoshGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
195 TanhGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
211 AsinhGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
224 AcoshGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
237 AtanhGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
251 SigmoidGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
267 SignGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
278 SinGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
291 CosGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
304 AsinGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
319 AcosGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
334 TanGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
347 AtanGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
363 BinaryGradCommon(const Scope& scope, const Operation& op, std::vector<Output>* grad_outputs, const Output& gx_1, const Output& gx_2) argument
376 AddGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
387 SubGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
399 MulGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
413 DivGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
428 RealDivGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
443 SquaredDifferenceGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
458 AddNGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
476 PowGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
513 MaximumMinimumGradCommon(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs, const Output& comparator) argument
529 MaximumGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
538 MinimumGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
547 RealGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
557 ImagGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
567 ComplexGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
576 AngleGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
592 ConjGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
699 SumGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
710 MeanGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
737 ErfGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
754 LgammaGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
766 MinOrMaxGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
832 ProdGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
1011 MatMulGradHelper(const Scope& scope, const bool is_batch, const Output& x0, const bool adj_x0, const Output& x1, const bool adj_x1, const Output& y0, const bool adj_y0, const Output& y1, const bool adj_y1, std::vector<Output>* grad_outputs) argument
1037 MatMulGradCommon(const Scope& scope, const Operation& op, const bool is_batch, const std::vector<Output>& grad_inputs, const string& attr_adj_x, const string& attr_adj_y, std::vector<Output>* grad_outputs) argument
1070 MatMulGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
1078 BatchMatMulGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
[all...] |
| /external/tensorflow/tensorflow/cc/framework/ |
| H A D | while_gradients.cc | 135 // returned in `grad_outputs`.
140 std::vector<Output>* grad_outputs) {
172 frame_name, grad_outputs,
181 std::vector<Output>* grad_outputs) {
195 scope, grad_outputs);
136 AddWhileGradientLoop(WhileContext* while_ctx, const std::vector<Output>& grad_inputs, const Output& backprop_execution_pred, const Scope& parent_scope, std::vector<Output>* grad_outputs) argument
179 AddWhileLoopGradient(WhileContext* while_ctx, const Scope& scope, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
|
| H A D | gradients_test.cc | 105 std::vector<Output> grad_outputs; local
107 AddSymbolicGradients(scope, {z}, {x, y}, {dz}, &grad_outputs));
132 std::vector<Output> grad_outputs; local
133 TF_ASSERT_OK(AddSymbolicGradients(scope, {z}, {x, y}, &grad_outputs));
164 std::vector<Output> grad_outputs; local
166 AddSymbolicGradients(scope, {z}, {u, v}, {dz}, &grad_outputs));
201 std::vector<Output> grad_outputs; local
203 &grad_outputs));
232 std::vector<Output> grad_outputs; local
234 {dx, dy, dz}, &grad_outputs));
266 std::vector<Output> grad_outputs; local
298 std::vector<Output> grad_outputs; local
329 std::vector<Output> grad_outputs; local
358 std::vector<Output> grad_outputs; local
395 std::vector<Output> grad_outputs; local
428 std::vector<Output> grad_outputs; local
458 std::vector<Output> grad_outputs; local
474 std::vector<Output> grad_outputs; local
492 std::vector<Output> grad_outputs; local
538 std::vector<Output> grad_outputs; local
547 EXPECT_EQ(NoGradient(), grad_outputs[0]); local
615 std::vector<Output> grad_outputs; local
624 EXPECT_EQ(NoGradient(), grad_outputs[0]); local
[all...] |
| H A D | gradients.cc | 55 std::vector<Output>* grad_outputs);
78 // Call the gradient function for `op`, storing the result in `grad_outputs`.
81 std::vector<Output>* grad_outputs);
143 const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs)
149 grad_outputs_(grad_outputs) {}
382 std::vector<Output>* grad_outputs) {
385 TF_RETURN_IF_ERROR(grad_fn(scope_, op, grad_inputs, grad_outputs));
561 std::vector<Output>* grad_outputs) {
563 inputs, grad_inputs, grad_outputs);
570 std::vector<Output>* grad_outputs) {
140 SymbolicGradientBuilder( const Scope& scope, const ops::GradOpRegistry* registry, const std::vector<Output>& outputs, const std::vector<Output>& inputs, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
379 CallGradFunction( const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
557 AddSymbolicGradients(const Scope& scope, const std::vector<Output>& outputs, const std::vector<Output>& inputs, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) argument
567 AddSymbolicGradients(const Scope& scope, const std::vector<Output>& outputs, const std::vector<Output>& inputs, std::vector<Output>* grad_outputs) argument
[all...] |
| /external/tensorflow/tensorflow/c/ |
| H A D | c_api_test.cc | 1455 TF_Output grad_outputs[2]; local
1461 AddGradients(grad_inputs_provided, inputs, 2, outputs, 1, grad_outputs);
1473 RunGraphsAndCompareOutputs(grad_outputs, expected_grad_outputs);
1479 TF_Output grad_outputs[1]; local
1483 AddGradients(grad_inputs_provided, inputs, 1, outputs, 1, grad_outputs);
1494 void RunGraphsAndCompareOutputs(TF_Output* grad_outputs, argument
1501 grad_outputs_vec.assign(grad_outputs, grad_outputs + 2);
1528 TF_Output* outputs, int noutputs, TF_Output* grad_outputs) {
1536 s_, grad_outputs);
1527 AddGradients(bool grad_inputs_provided, TF_Output* inputs, int ninputs, TF_Output* outputs, int noutputs, TF_Output* grad_outputs) argument
[all...] |