Searched refs:first_zero_index (Results 1 – 1 of 1) sorted by relevance
666 const auto first_zero_index = std::get<1>(mask.max(dim, /*keepdim*/ true)); in cumprod_backward() local668 .scatter_(dim, first_zero_index, /*src*/ 1) in cumprod_backward()681 … .mul_(at::gather(output_conj, dim, (first_zero_index - 1).relu_()) in cumprod_backward()682 .masked_fill_(first_zero_index == 0, 1.)) in cumprod_backward()