Searched defs:prods_until_k (Results 1 – 1 of 1) sorted by relevance
726 const Tensor prods_until_k = at::prod(input_conj.slice(dim, 0, k), dim, true); in cumprod_backward() local729 const Tensor prods_until_k = at::prod(input_conj.slice(dim, 0, k), dim, true); in cumprod_backward() local