Home
last modified time | relevance | path

Searched defs:grad_outputs (Results 1 – 25 of 40) sorted by relevance

12

/external/tensorflow/tensorflow/cc/gradients/
Darray_grad.cc43 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 SplitVGrad()
148 std::vector<Output>* grad_outputs) { in FillGrad()
[all …]
Dmath_grad.cc59 std::vector<Output>* grad_outputs) { in AbsGrad()
68 std::vector<Output>* grad_outputs) { in NegGrad()
77 std::vector<Output>* grad_outputs) { in InvGrad()
88 std::vector<Output>* grad_outputs) { in SquareGrad()
101 std::vector<Output>* grad_outputs) { in SqrtGrad()
111 std::vector<Output>* grad_outputs) { in RsqrtGrad()
121 std::vector<Output>* grad_outputs) { in ExpGrad()
133 std::vector<Output>* grad_outputs) { in Expm1Grad()
146 std::vector<Output>* grad_outputs) { in LogGrad()
159 std::vector<Output>* grad_outputs) { in Log1pGrad()
[all …]
Dnn_grad.cc29 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 …]
Dimage_grad.cc34 std::vector<Output>* grad_outputs) { in ResizeNearestNeighborGradHelper()
56 std::vector<Output>* grad_outputs) { in ResizeBilinearGradHelper()
74 std::vector<Output>* grad_outputs) { in ResizeBicubicGradHelper()
93 std::vector<Output>* grad_outputs) { in ScaleAndTranslateGradHelper()
114 std::vector<Output>* grad_outputs) { in CropAndResizeGradHelper()
Ddata_flow_grad.cc46 std::vector<Output>* grad_outputs) { in DynamicPartitionGrad()
111 std::vector<Output>* grad_outputs) { in DynamicStitchGrad()
Dresource_variable_grad.cc28 std::vector<Output>* grad_outputs) { in ReadVariableOpGrad()
Dgrad_testutil.cc25 std::vector<Output>* grad_outputs) { in CallGradFunction()
Dmanip_grad.cc27 std::vector<Output>* grad_outputs) { in RollGrad()
Dfunctional_grad.cc29 std::vector<Output>* grad_outputs) { in PartitionedCallGrad()
Dlinalg_grad.cc361 std::vector<Output>* grad_outputs) { in EinsumGrad()
/external/tensorflow/tensorflow/cc/framework/
Dgradients_test.cc105 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 …]
Dgradients.cc144 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()
Dwhile_gradients.cc140 std::vector<Output>* grad_outputs) { in AddWhileGradientLoop()
181 std::vector<Output>* grad_outputs) { in AddWhileLoopGradient()
/external/tensorflow/tensorflow/c/experimental/gradients/
Dmath_grad.cc54 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
75 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
99 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
125 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
221 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
240 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
277 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
324 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
395 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
Dnn_grad.cc48 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
100 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
124 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
Dnot_differentiable.cc20 AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
Darray_grad.cc25 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
Dcustom_gradient_test.cc48 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
/external/pytorch/torch/csrc/autograd/
Dautograd.cpp28 const variable_list& grad_outputs) { in _make_grads()
92 const variable_list& grad_outputs, in run_backward()
187 const variable_list& grad_outputs, in grad()
Dpython_hook.cpp143 THPObjectPtr grad_outputs(wrap_variables(grad_outputs_)); in operator ()() local
168 THPObjectPtr grad_outputs(wrap_variables(_inputs)); in operator ()() local
/external/pytorch/torch/nn/parallel/
D_functions.py32 def backward(ctx, *grad_outputs):
49 def backward(ctx, *grad_outputs):
/external/pytorch/test/cpp_extensions/
Didentity.cpp12 static tensor_list backward(AutogradContext* ctx, tensor_list grad_outputs) { in backward()
/external/pytorch/torch/utils/
D__init__.py64 def error_pre_hook(grad_outputs): argument
/external/pytorch/torch/distributed/nn/
Dfunctional.py267 def backward(ctx, *grad_outputs):
333 def backward(ctx, *grad_outputs):
399 def backward(ctx, *grad_outputs):
/external/pytorch/torch/csrc/api/src/nn/modules/
D_functions.cpp87 variable_list grad_outputs) { in backward()

12