Searched defs:grad_slice (Results 1 – 3 of 3) sorted by relevance
97 at::Tensor grad_slice; in apply_impl() local
740 auto grad_slice = at::sum(grad.slice(dim, k) * omitted_products, dim); in cumprod_backward() local
3276 auto grad_slice = grad_.as_strided_symint(sizes, strides, storage_offset); in as_strided_scatter_backward() local