| /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 SplitVGrad()
147 const std::vector<Output>& grad_inputs, in FillGrad()
[all …]
|
| D | math_grad.cc | 58 const std::vector<Output>& grad_inputs, in AbsGrad()
67 const std::vector<Output>& grad_inputs, in NegGrad()
76 const std::vector<Output>& grad_inputs, in InvGrad()
87 const std::vector<Output>& grad_inputs, in SquareGrad()
100 const std::vector<Output>& grad_inputs, in SqrtGrad()
110 const std::vector<Output>& grad_inputs, in RsqrtGrad()
120 const std::vector<Output>& grad_inputs, in ExpGrad()
132 const std::vector<Output>& grad_inputs, in Expm1Grad()
145 const std::vector<Output>& grad_inputs, in LogGrad()
158 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 | 33 const std::vector<Output>& grad_inputs, in ResizeNearestNeighborGradHelper()
55 const std::vector<Output>& grad_inputs, in ResizeBilinearGradHelper()
73 const std::vector<Output>& grad_inputs, in ResizeBicubicGradHelper()
92 const std::vector<Output>& grad_inputs, in ScaleAndTranslateGradHelper()
113 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 | resource_variable_grad.cc | 27 const std::vector<Output>& grad_inputs, in ReadVariableOpGrad()
|
| 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()
|
| D | functional_grad.cc | 28 const std::vector<Output>& grad_inputs, in PartitionedCallGrad()
|
| D | linalg_grad.cc | 360 const std::vector<Output>& grad_inputs, in EinsumGrad()
|
| /external/tensorflow/tensorflow/c/experimental/gradients/ |
| D | math_grad.cc | 55 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
76 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
100 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
126 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
222 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
241 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
278 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
325 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
396 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
|
| D | nn_grad.cc | 49 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
101 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
125 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 | 1680 TF_Output grad_inputs[1]; in AddGradients() local
|