| /external/tensorflow/tensorflow/cc/gradients/ |
| 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 FillGrad()
152 std::vector<Output>* grad_outputs) { in DiagGrad()
[all …]
|
| D | math_grad.cc | 57 std::vector<Output>* grad_outputs) { in AbsGrad()
66 std::vector<Output>* grad_outputs) { in NegGrad()
75 std::vector<Output>* grad_outputs) { in InvGrad()
86 std::vector<Output>* grad_outputs) { in SquareGrad()
99 std::vector<Output>* grad_outputs) { in SqrtGrad()
109 std::vector<Output>* grad_outputs) { in RsqrtGrad()
119 std::vector<Output>* grad_outputs) { in ExpGrad()
131 std::vector<Output>* grad_outputs) { in Expm1Grad()
144 std::vector<Output>* grad_outputs) { in LogGrad()
157 std::vector<Output>* grad_outputs) { in Log1pGrad()
[all …]
|
| 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 …]
|
| D | image_grad.cc | 28 std::vector<Output>* grad_outputs) { in ResizeNearestNeighborGradHelper()
50 std::vector<Output>* grad_outputs) { in ResizeBilinearGradHelper()
68 std::vector<Output>* grad_outputs) { in ResizeBicubicGradHelper()
87 std::vector<Output>* grad_outputs) { in ScaleAndTranslateGradHelper()
108 std::vector<Output>* grad_outputs) { in CropAndResizeGradHelper()
|
| D | data_flow_grad.cc | 46 std::vector<Output>* grad_outputs) { in DynamicPartitionGrad()
111 std::vector<Output>* grad_outputs) { in DynamicStitchGrad()
|
| D | grad_testutil.cc | 25 std::vector<Output>* grad_outputs) { in CallGradFunction()
|
| D | manip_grad.cc | 27 std::vector<Output>* grad_outputs) { in RollGrad()
|
| D | math_grad_test.cc | 895 std::vector<Output> grad_outputs; in TEST_F() local
|
| /external/tensorflow/tensorflow/cc/framework/ |
| 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 …]
|
| 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()
|
| D | while_gradients.cc | 140 std::vector<Output>* grad_outputs) { in AddWhileGradientLoop()
181 std::vector<Output>* grad_outputs) { in AddWhileLoopGradient()
|
| /external/tensorflow/tensorflow/c/experimental/gradients/ |
| D | math_grad.cc | 41 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
62 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
87 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
113 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
225 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
244 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
281 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
328 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
401 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
|
| D | nn_grad.cc | 48 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
101 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
125 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
|
| D | array_grad.cc | 25 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
|
| D | custom_gradient_test.cc | 48 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_test.cc | 1572 TF_Output grad_outputs[2]; in TestGradientsSuccess() local
1596 TF_Output grad_outputs[1]; in TestGradientsError() local
1612 void RunGraphsAndCompareOutputs(TF_Output* grad_outputs, in RunGraphsAndCompareOutputs()
1647 int noutputs, TF_Output* grad_outputs) { in AddGradients()
1807 TF_Output grad_outputs[2]; in BuildGraphAndAddGradientsWithPrefixes() local
|