| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_fft_test.cc | 23 EXPECT_EQ(::xla::FftType::FFT, in TEST()
24 static_cast<::tensorflow::int32>(::tensorflow::xla::FftType::FFT)); in TEST()
25 EXPECT_EQ(::xla::FftType::IFFT, in TEST()
26 static_cast<::tensorflow::int32>(::tensorflow::xla::FftType::IFFT)); in TEST()
27 EXPECT_EQ(::xla::FftType::RFFT, in TEST()
28 static_cast<::tensorflow::int32>(::tensorflow::xla::FftType::RFFT)); in TEST()
29 EXPECT_EQ(::xla::FftType::IRFFT, static_cast<::tensorflow::int32>( in TEST()
30 ::tensorflow::xla::FftType::IRFFT)); in TEST()
|
| D | runtime_fft_impl.h | 30 enum class FftType : int32 { enum
181 FftType fft_type, bool double_precision, in EigenFftWithRank()
185 case FftType::FFT: in EigenFftWithRank()
198 case FftType::IFFT: in EigenFftWithRank()
211 case FftType::RFFT: in EigenFftWithRank()
223 case FftType::IRFFT: in EigenFftWithRank()
245 FftType fft_type, bool double_precision, int32 fft_rank, in EigenFftImpl()
|
| D | runtime_single_threaded_fft.cc | 30 static_cast<tensorflow::xla::FftType>(fft_type), in __xla_cpu_runtime_EigenSingleThreadedFft()
|
| D | runtime_fft.cc | 38 static_cast<tensorflow::xla::FftType>(fft_type), in __xla_cpu_runtime_EigenFft()
|
| /external/webrtc/modules/audio_processing/utility/ |
| D | pffft_wrapper_unittest.cc | 30 void CreatePffftWrapper(size_t fft_size, Pffft::FftType fft_type) { in CreatePffftWrapper()
69 Pffft::FftType fft_type = in PffftValidateWrapper()
70 complex_fft ? Pffft::FftType::kComplex : Pffft::FftType::kReal; in PffftValidateWrapper()
117 if (Pffft::IsValidFftSize(fft_size, Pffft::FftType::kReal)) { in TEST()
118 CreatePffftWrapper(fft_size, Pffft::FftType::kReal); in TEST()
120 if (Pffft::IsValidFftSize(fft_size, Pffft::FftType::kComplex)) { in TEST()
121 CreatePffftWrapper(fft_size, Pffft::FftType::kComplex); in TEST()
134 ASSERT_FALSE(Pffft::IsValidFftSize(fft_size, Pffft::FftType::kReal)); in TEST_P()
135 EXPECT_DEATH(CreatePffftWrapper(fft_size, Pffft::FftType::kReal), ""); in TEST_P()
140 ASSERT_FALSE(Pffft::IsValidFftSize(fft_size, Pffft::FftType::kComplex)); in TEST_P()
[all …]
|
| D | pffft_wrapper.h | 27 enum class FftType { kReal, kComplex }; enum
42 FloatBuffer(size_t fft_size, FftType fft_type);
56 Pffft(size_t fft_size, FftType fft_type);
62 static bool IsValidFftSize(size_t fft_size, FftType fft_type);
87 const FftType fft_type_;
|
| D | pffft_wrapper.cc | 19 size_t GetBufferSize(size_t fft_size, Pffft::FftType fft_type) { in GetBufferSize()
20 return fft_size * (fft_type == Pffft::FftType::kReal ? 1 : 2); in GetBufferSize()
29 Pffft::FloatBuffer::FloatBuffer(size_t fft_size, FftType fft_type) in FloatBuffer()
45 Pffft::Pffft(size_t fft_size, FftType fft_type) in Pffft()
50 fft_type == Pffft::FftType::kReal ? PFFFT_REAL : PFFFT_COMPLEX)), in Pffft()
62 bool Pffft::IsValidFftSize(size_t fft_size, FftType fft_type) { in IsValidFftSize()
78 int a_min = (fft_type == Pffft::FftType::kReal) ? 5 : 4; in IsValidFftSize()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | fft_ops.cc | 38 using xla::FftType;
42 explicit GenericFftOp(OpKernelConstruction* ctx, FftType fft_type, in GenericFftOp()
54 if (fft_type_ == FftType::RFFT || fft_type_ == FftType::IRFFT) { in Compile()
65 if (fft_type_ == FftType::IRFFT) { in Compile()
103 const FftType fft_type_;
114 : GenericFftOp(ctx, /*fft_type=*/FftType::FFT, /*fft_rank=*/FFTRank) {} in FFTOp()
126 : GenericFftOp(ctx, /*fft_type=*/FftType::IFFT, /*fft_rank=*/FFTRank) {} in IFFTOp()
139 : GenericFftOp(ctx, /*fft_type=*/FftType::RFFT, /*fft_rank=*/FFTRank) {} in RFFTOp()
161 : GenericFftOp(ctx, /*fft_type=*/FftType::IRFFT, /*fft_rank=*/FFTRank) {} in IRFFTOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | attribute_exporter.cc | 136 StatusOr<FftType> ConvertFftType(llvm::StringRef type_string) { in ConvertFftType()
137 llvm::Optional<mlir::mhlo::FftType> type = in ConvertFftType()
138 mlir::mhlo::symbolizeEnum<mlir::mhlo::FftType>(type_string); in ConvertFftType()
142 case mlir::mhlo::FftType::FFT: in ConvertFftType()
143 return xla::FftType::FFT; in ConvertFftType()
144 case mlir::mhlo::FftType::IFFT: in ConvertFftType()
145 return xla::FftType::IFFT; in ConvertFftType()
146 case mlir::mhlo::FftType::RFFT: in ConvertFftType()
147 return xla::FftType::RFFT; in ConvertFftType()
148 case mlir::mhlo::FftType::IRFFT: in ConvertFftType()
[all …]
|
| D | attribute_importer.cc | 127 StatusOr<mlir::mhlo::FftType> ConvertFftType(FftType type) { in ConvertFftType()
129 case FftType::FFT: in ConvertFftType()
130 return mlir::mhlo::FftType::FFT; in ConvertFftType()
131 case FftType::IFFT: in ConvertFftType()
132 return mlir::mhlo::FftType::IFFT; in ConvertFftType()
133 case FftType::RFFT: in ConvertFftType()
134 return mlir::mhlo::FftType::RFFT; in ConvertFftType()
135 case FftType::IRFFT: in ConvertFftType()
136 return mlir::mhlo::FftType::IRFFT; in ConvertFftType()
|
| D | attribute_importer.h | 48 StatusOr<mlir::mhlo::FftType> ConvertFftType(FftType type);
|
| D | attribute_exporter.h | 43 StatusOr<FftType> ConvertFftType(llvm::StringRef type_string);
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | fft_thunk.cc | 63 se::fft::Type FftTypeToSeType(FftType type, bool double_precision) { in FftTypeToSeType()
65 case FftType::FFT: in FftTypeToSeType()
68 case FftType::IFFT: in FftTypeToSeType()
71 case FftType::IRFFT: in FftTypeToSeType()
73 case FftType::RFFT: in FftTypeToSeType()
101 FftThunk::FftThunk(ThunkInfo thunk_info, FftType fft_type, in FftThunk()
|
| D | fft_thunk.h | 66 FftThunk(ThunkInfo thunk_info, FftType fft_type,
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_compiler.cc | 580 py::enum_<FftType>(m, "FftType") in BuildXlaCompilerSubmodule()
581 .value("FFT", FftType::FFT) in BuildXlaCompilerSubmodule()
582 .value("IFFT", FftType::IFFT) in BuildXlaCompilerSubmodule()
583 .value("RFFT", FftType::RFFT) in BuildXlaCompilerSubmodule()
584 .value("IRFFT", FftType::IRFFT); in BuildXlaCompilerSubmodule()
|
| D | xla_client.py | 422 FftType = _xla.FftType variable
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | hlo_ops_base_enums.td | 38 // These mirror the XLA FftType proto enum.
44 def HLO_FftTypeAttr : StrEnumAttr<"FftType",
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 999 void Sweep(int64 fft_rank, FftType fft_type, in Sweep()
1003 const bool inverse = fft_type == FftType::IFFT || fft_type == FftType::IRFFT; in Sweep()
1004 const bool input_is_truncated = fft_type == FftType::IRFFT; in Sweep()
1005 const bool output_is_truncated = fft_type == FftType::RFFT; in Sweep()
1131 int64 fft_rank, FftType fft_type, int64 fft_size, in CopyDataFromInput()
1139 const bool input_is_truncated = fft_type == FftType::IRFFT; in CopyDataFromInput()
1184 int64 fft_rank, FftType fft_type, in CopyDataToOutput()
1190 const bool output_is_truncated = fft_type == FftType::RFFT; in CopyDataToOutput()
1221 int64 fft_rank, FftType fft_type, int64 fft_size, in CopyDataFromInput()
1227 const bool input_is_float = fft_type == FftType::RFFT; in CopyDataFromInput()
[all …]
|
| D | shape_inference_test.cc | 810 static void Pass(const Shape& shape, FftType type, in Pass()
818 static void Fail(const Shape& shape, FftType type, in Fail()
829 FftType type = FftType::FFT; in TEST_F()
839 FftType type = FftType::FFT; in TEST_F()
847 FftType type = FftType::IFFT; in TEST_F()
857 FftType type = FftType::IFFT; in TEST_F()
865 FftType type = FftType::RFFT; in TEST_F()
876 FftType type = FftType::RFFT; in TEST_F()
896 FftType type = FftType::RFFT; in TEST_F()
904 FftType type = FftType::IRFFT; in TEST_F()
[all …]
|
| D | shape_inference.h | 118 static StatusOr<Shape> InferFftShape(const Shape& in, FftType fft_type,
|
| D | hlo_instructions.h | 107 FftType fft_type,
109 FftType fft_type() const { return fft_type_; } in fft_type()
130 FftType fft_type_ = FftType::FFT;
|
| /external/webrtc/modules/audio_processing/agc2/rnn_vad/ |
| D | auto_correlation.cc | 29 : fft_(1 << kAutoCorrelationFftOrder, Pffft::FftType::kReal), in AutoCorrelationCalculator()
|
| D | spectral_features_internal_unittest.cc | 92 Pffft fft(kFrameSize20ms24kHz, Pffft::FftType::kReal); in TEST()
|
| D | spectral_features.cc | 90 fft_(kFrameSize20ms24kHz, Pffft::FftType::kReal), in SpectralFeaturesExtractor()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.h | 124 FftType fft_type,
|