Searched refs:neg_sizes (Results 1 – 2 of 2) sorted by relevance
138 auto neg_sizes = in_dims; in EigenFftC2R() local139 neg_sizes[FFTRank] = fft_shape[FFTRank - 1] - in_dims[FFTRank]; in EigenFftC2R()167 if (neg_sizes[FFTRank] != 0) { in EigenFftC2R()168 full_fft.slice(neg_target_indices, neg_sizes).device(device) = in EigenFftC2R()169 full_fft.slice(neg_start_indices, neg_sizes) in EigenFftC2R()
272 auto neg_sizes = input_slice_sizes; in DoRealBackwardFFT() local273 neg_sizes[FFTRank] = fft_shape[FFTRank - 1] - input_slice_sizes[FFTRank]; in DoRealBackwardFFT()302 if (neg_sizes[FFTRank] != 0) { in DoRealBackwardFFT()303 full_fft.slice(neg_target_indices, neg_sizes).device(device) = in DoRealBackwardFFT()304 full_fft.slice(neg_start_indices, neg_sizes) in DoRealBackwardFFT()