Searched refs:src_zero (Results 1 – 11 of 11) sorted by relevance
15 let src_zero = F::Int::ZERO; in trunc() localVariable90 let sticky = if (significand << (src_bits - shift)) != src_zero { in trunc()93 src_zero in trunc()
15 let src_zero = F::Int::ZERO; in extend() localVariable57 } else if a_abs != src_zero { in extend()
6840 Tensor src_zero = src == 0; in scatter_reduce_backward() local6843 .scatter_add(dim, index, src_zero.to(self.dtype())) in scatter_reduce_backward()6845 Tensor src_single_zero = bitwise_and(src_zero, src_num_zeros == 1); in scatter_reduce_backward()6855 (grad * result).gather(dim, index) / src.masked_fill(src_zero, 1)); in scatter_reduce_backward()6936 Tensor src_zero = source == 0; in index_reduce_backward() local6938 .index_add(dim, index, src_zero.to(self.dtype())) in index_reduce_backward()6940 Tensor src_single_zero = bitwise_and(src_zero, src_num_zeros == 1); in index_reduce_backward()6951 source.masked_fill(src_zero, 1)); in index_reduce_backward()