| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad.cc | 42 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 FillGrad()
151 const std::vector<Output>& grad_inputs, in DiagGrad()
[all …]
|
| D | math_grad.cc | 56 const std::vector<Output>& grad_inputs, in AbsGrad()
65 const std::vector<Output>& grad_inputs, in NegGrad()
74 const std::vector<Output>& grad_inputs, in InvGrad()
85 const std::vector<Output>& grad_inputs, in SquareGrad()
98 const std::vector<Output>& grad_inputs, in SqrtGrad()
108 const std::vector<Output>& grad_inputs, in RsqrtGrad()
118 const std::vector<Output>& grad_inputs, in ExpGrad()
130 const std::vector<Output>& grad_inputs, in Expm1Grad()
143 const std::vector<Output>& grad_inputs, in LogGrad()
156 const std::vector<Output>& grad_inputs, in Log1pGrad()
[all …]
|
| D | nn_grad.cc | 28 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 …]
|
| D | image_grad.cc | 27 const std::vector<Output>& grad_inputs, in ResizeNearestNeighborGradHelper()
49 const std::vector<Output>& grad_inputs, in ResizeBilinearGradHelper()
67 const std::vector<Output>& grad_inputs, in ResizeBicubicGradHelper()
86 const std::vector<Output>& grad_inputs, in ScaleAndTranslateGradHelper()
107 const std::vector<Output>& grad_inputs, in CropAndResizeGradHelper()
|
| D | data_flow_grad.cc | 45 const std::vector<Output>& grad_inputs, in DynamicPartitionGrad()
110 const std::vector<Output>& grad_inputs, in DynamicStitchGrad()
|
| D | grad_testutil.cc | 24 const std::vector<Output>& grad_inputs, in CallGradFunction()
|
| D | manip_grad.cc | 26 const std::vector<Output>& grad_inputs, in RollGrad()
|
| /external/tensorflow/tensorflow/c/experimental/gradients/ |
| D | math_grad.cc | 42 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
63 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
88 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
114 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
226 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
245 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
282 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
329 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
402 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
|
| D | nn_grad.cc | 49 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
102 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
126 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
|
| D | not_differentiable.cc | 21 absl::Span<AbstractTensorHandle*> grad_inputs) { in Compute()
|
| D | array_grad.cc | 26 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
|
| D | custom_gradient_test.cc | 49 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | gradients.cc | 144 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
|
| D | while_gradients.cc | 137 const std::vector<Output>& grad_inputs, in AddWhileGradientLoop()
180 const std::vector<Output>& grad_inputs, in AddWhileLoopGradient()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_test.cc | 1649 TF_Output grad_inputs[1]; in AddGradients() local
|