Home
last modified time | relevance | path

Searched defs:grad_inputs (Results 1 – 25 of 29) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/
H A Dmath_grad.cc58 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 …]
H A Darray_grad.cc42 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 …]
H A Dnn_grad.cc28 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 …]
H A Dimage_grad.cc33 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()
H A Ddata_flow_grad.cc45 const std::vector<Output>& grad_inputs, in DynamicPartitionGrad()
110 const std::vector<Output>& grad_inputs, in DynamicStitchGrad()
H A Dresource_variable_grad.cc27 const std::vector<Output>& grad_inputs, in ReadVariableOpGrad()
H A Dgrad_testutil.cc24 const std::vector<Output>& grad_inputs, in CallGradFunction()
H A Dmanip_grad.cc26 const std::vector<Output>& grad_inputs, in RollGrad()
H A Dfunctional_grad.cc28 const std::vector<Output>& grad_inputs, in PartitionedCallGrad()
H A Dlinalg_grad.cc360 const std::vector<Output>& grad_inputs, in EinsumGrad()
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/gradients/
H A Dmath_grad.cc55 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()
H A Dnn_grad.cc49 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()
H A Dnot_differentiable.cc21 absl::Span<AbstractTensorHandle*> grad_inputs) { in Compute()
H A Darray_grad.cc26 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
H A Dcustom_gradient_test.cc49 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
/aosp_15_r20/external/tensorflow/tensorflow/cc/framework/
H A Dgradients.cc144 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
H A Dwhile_gradients.cc137 const std::vector<Output>& grad_inputs, in AddWhileGradientLoop()
180 const std::vector<Output>& grad_inputs, in AddWhileLoopGradient()
/aosp_15_r20/external/pytorch/torch/csrc/autograd/functions/
H A Dtensor.cpp22 variable_list grad_inputs(2); in apply() local
154 variable_list grad_inputs(num_outputs()); in apply_impl() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dautodiff.cpp167 auto grad_inputs = insertGraph(*graph, *bw_graph, grad); in build_script_grad() local
433 value_list grad_inputs = in addReverseInline() local
/aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/
H A Dts_autograd_functions.cpp55 torch::autograd::variable_list grad_inputs = { in backward() local
/aosp_15_r20/external/pytorch/torch/distributed/pipelining/
H A D_backward.py186 def hook(grad_inputs): argument
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dautograd.cpp140 variable_list grad_inputs = Engine::get_default_engine().execute( in run_backward() local
H A Dpython_hook.cpp167 THPObjectPtr grad_inputs(wrap_variables(_outputs)); in operator ()() local
H A DFunctionsManual.cpp1075 std::vector<Tensor> grad_inputs(sizes.size()); in cat_tensors_backward() local
1114 std::vector<Tensor> grad_inputs(dtypes.size()); in stack_tensors_backward() local
1133 std::vector<Tensor> grad_inputs(sizes.size()); in block_diag_backward() local
6760 std::tuple<Tensor, Tensor, Tensor> grad_inputs = in _cudnn_convolution_backward() local
/aosp_15_r20/external/pytorch/test/
H A Dtest_foreach.py1445 def hook(grad_inputs, grad_outputs) -> None: argument

12