Searched refs:double_precision (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_fft_impl.h | 181 FftType fft_type, bool double_precision, in EigenFftWithRank() argument
186 if (double_precision) { in EigenFftWithRank()
199 if (double_precision) { in EigenFftWithRank()
212 if (double_precision) { in EigenFftWithRank()
224 if (double_precision) { in EigenFftWithRank()
245 FftType fft_type, bool double_precision, int32 fft_rank, in EigenFftImpl() argument
251 double_precision, input_batch, in EigenFftImpl()
256 double_precision, input_batch, in EigenFftImpl()
261 device, out, operand, fft_type, double_precision, input_batch, in EigenFftImpl()
|
| D | runtime_single_threaded_fft.cc | 27 int32 double_precision, int32 fft_rank, int64 input_batch, in __xla_cpu_runtime_EigenSingleThreadedFft() argument
31 static_cast<bool>(double_precision), fft_rank, in __xla_cpu_runtime_EigenSingleThreadedFft()
|
| D | runtime_fft.cc | 31 int32 double_precision, int32 fft_rank, int64 input_batch, in __xla_cpu_runtime_EigenFft() argument
39 static_cast<bool>(double_precision), fft_rank, input_batch, fft_length0, in __xla_cpu_runtime_EigenFft()
|
| D | runtime_fft.h | 26 tensorflow::int32 double_precision, tensorflow::int32 fft_rank,
|
| D | runtime_single_threaded_fft.h | 26 tensorflow::int32 double_precision, tensorflow::int32 fft_rank,
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | fft_thunk.cc | 63 se::fft::Type FftTypeToSeType(FftType type, bool double_precision) { in FftTypeToSeType() argument
66 return double_precision ? se::fft::Type::kZ2ZForward in FftTypeToSeType()
69 return double_precision ? se::fft::Type::kZ2ZInverse in FftTypeToSeType()
72 return double_precision ? se::fft::Type::kZ2D : se::fft::Type::kC2R; in FftTypeToSeType()
74 return double_precision ? se::fft::Type::kD2Z : se::fft::Type::kR2C; in FftTypeToSeType()
|