Home
last modified time | relevance | path

Searched defs:grad_in (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DGridSampler.h43 bool align_corners, scalar_t *grad_in) { in grid_sampler_unnormalize_set_grad()
66 scalar_t *grad_in) { in clip_coordinates_set_grad()
112 int64_t twice_high, scalar_t *grad_in) { in reflect_coordinates_set_grad()
183 scalar_t *grad_in) { in grid_sampler_compute_source_index_set_grad()
H A DUnfoldBackward.h34 const Tensor& grad_in, in _make_unfold_backward_iter_over_grad_out()
H A DReduceOps.cpp2301 auto grad_in = at::zeros_symint(sizes, grad_out.options()); in value_selecting_reduction_backward_symint() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DGridSampler.cuh39 bool align_corners, scalar_t *grad_in) { in grid_sampler_unnormalize_set_grad()
63 scalar_t clip_coordinates_set_grad(scalar_t in, int clip_limit, scalar_t *grad_in) { in clip_coordinates_set_grad()
110 scalar_t *grad_in) { in reflect_coordinates_set_grad()
195 scalar_t *grad_in) { in grid_sampler_compute_source_index_set_grad()
H A DUnfoldBackwardKernel.cu123 const Tensor& grad_in, in unfold_backward_cuda_kernel()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DUnfoldBackwardKernel.cpp113 const Tensor& grad_in, in unfold_backward_cpu_kernel()
H A DPaddingKernel.cpp115 static inline void add_stub(scalar_t* grad_in, const scalar_t* grad_out, int64_t size) { in add_stub()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dmaxpooling_op.cc1117 const Tensor& grad_in = context->input(1); in Compute() local
1185 const Tensor& grad_in = context->input(1); in Compute() local
/aosp_15_r20/external/pytorch/test/nn/
H A Dtest_module_hooks.py1336 def bw_hook(m, grad_in, grad_output): argument
/aosp_15_r20/external/pytorch/test/
H A Dtest_nn.py395 def hook(self, module, grad_out, grad_in): argument
H A Dtest_autograd.py1395 def acc_grad_node_post_hook(grad_in, grad_out): argument