| /external/tensorflow/tensorflow/cc/gradients/ |
| H A D | grad_testutil.cc | 24 const std::vector<Output>& grad_inputs,
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 | 45 const std::vector<Output>& grad_inputs,
99 DynamicStitch(scope, partitioned_indices.outputs, grad_inputs);
110 const std::vector<Output>& grad_inputs,
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 | 28 const std::vector<Output>& grad_inputs,
41 auto dyy = Mul(scope, grad_inputs[0], y);
43 auto sub = Sub(scope, grad_inputs[0], sum);
51 const std::vector<Output>& grad_inputs,
54 auto sum = Sum(scope, grad_inputs[0], {1}, Sum::KeepDims(true));
56 auto dx = Sub(scope, grad_inputs[0], mul);
63 const std::vector<Output>& grad_inputs,
65 auto dx = internal::ReluGrad(scope, grad_inputs[0], op.input(0));
72 const std::vector<Output>& grad_inputs,
74 auto dx = internal::Relu6Grad(scope, grad_inputs[
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 | 43 const std::vector<Output>& grad_inputs,
51 auto grad_op = Unstack(scope, grad_inputs[0], N, Unstack::Axis(axis));
60 const std::vector<Output>& grad_inputs,
64 grad_outputs->push_back(Stack(scope, grad_inputs, Stack::Axis(axis)));
70 const std::vector<Output>& grad_inputs,
72 grad_outputs->push_back(Identity(scope, grad_inputs[0]));
78 const std::vector<Output>& grad_inputs,
80 grad_outputs->push_back(Identity(scope, grad_inputs[0]));
86 const std::vector<Output>& grad_inputs,
88 grad_outputs->push_back(Identity(scope, grad_inputs[
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 | 56 const std::vector<Output>& grad_inputs,
59 grad_outputs->push_back(Mul(scope, grad_inputs[0], Sign(scope, op.input(0))));
65 const std::vector<Output>& grad_inputs,
68 grad_outputs->push_back(Neg(scope, grad_inputs[0]));
74 const std::vector<Output>& grad_inputs,
78 internal::ReciprocalGrad(scope, op.output(0), grad_inputs[0]));
85 const std::vector<Output>& grad_inputs,
92 Mul(scope, grad_inputs[0], ConjugateHelper(scope, dydx)));
98 const std::vector<Output>& grad_inputs,
102 internal::SqrtGrad(scope, op.output(0), grad_inputs[
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
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
662 SumGradHelper(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs) 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
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 | 131 // associated with `while_ctx`. `grad_inputs` are the partial derivatives
137 const std::vector<Output>& grad_inputs,
141 DCHECK_EQ(grad_inputs.size(), while_ctx->body_outputs().size());
171 TF_RETURN_IF_ERROR(BuildWhileLoop(scope, grad_inputs, cond_fn, body_fn,
180 const std::vector<Output>& grad_inputs,
194 return AddWhileGradientLoop(while_ctx, grad_inputs, backprop_counter_cond,
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.cc | 54 const std::vector<Output>& grad_inputs,
80 const std::vector<Output>& grad_inputs,
143 const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs)
148 grad_inputs_(grad_inputs),
381 const std::vector<Output>& grad_inputs,
385 TF_RETURN_IF_ERROR(grad_fn(scope_, op, grad_inputs, grad_outputs));
560 const std::vector<Output>& grad_inputs,
563 inputs, grad_inputs, grad_outputs);
571 std::vector<Output> grad_inputs; local
572 grad_inputs
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
[all...] |
| /external/tensorflow/tensorflow/c/ |
| H A D | c_api_test.cc | 1530 TF_Output grad_inputs[1]; local
1534 grad_inputs[0] = TF_Output{grad_inputs_op, 0};
1535 TF_AddGradients(graph_, outputs, noutputs, inputs, ninputs, grad_inputs,
|