Home
last modified time | relevance | path

Searched refs:zero_start_indices (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_fft_impl.h97 const Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> zero_start_indices; in EigenFftR2C() local
102 output.device(device) = full_fft.slice(zero_start_indices, out_dims); in EigenFftR2C()
143 const Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> zero_start_indices; in EigenFftC2R() local
147 full_fft.slice(zero_start_indices, in_dims).device(device) = input; in EigenFftC2R()
155 full_fft.slice(zero_start_indices, in_dims).device(device) = in EigenFftC2R()
156 full_fft.slice(zero_start_indices, in_dims) in EigenFftC2R()
/external/tensorflow/tensorflow/core/kernels/
Dfft_ops.cc226 const Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> zero_start_indices; in DoRealForwardFFT() local
234 input.slice(zero_start_indices, input_slice_sizes) in DoRealForwardFFT()
239 full_fft.slice(zero_start_indices, output.dimensions()); in DoRealForwardFFT()