Searched defs:grad_in (Results 1 – 11 of 11) sorted by relevance
/aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
H A D | GridSampler.h | 43 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 D | UnfoldBackward.h | 34 const Tensor& grad_in, in _make_unfold_backward_iter_over_grad_out()
|
H A D | ReduceOps.cpp | 2301 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 D | GridSampler.cuh | 39 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 D | UnfoldBackwardKernel.cu | 123 const Tensor& grad_in, in unfold_backward_cuda_kernel()
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
H A D | UnfoldBackwardKernel.cpp | 113 const Tensor& grad_in, in unfold_backward_cpu_kernel()
|
H A D | PaddingKernel.cpp | 115 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 D | maxpooling_op.cc | 1117 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 D | test_module_hooks.py | 1336 def bw_hook(m, grad_in, grad_output): argument
|
/aosp_15_r20/external/pytorch/test/ |
H A D | test_nn.py | 395 def hook(self, module, grad_out, grad_in): argument
|
H A D | test_autograd.py | 1395 def acc_grad_node_post_hook(grad_in, grad_out): argument
|