Adds FftType and FftRequest in anticipation of HLO plumbing for FFT ops.

PiperOrigin-RevId: 174356536

1 files changed, 16 insertions, 1 deletions

diff --git a/tensorflow/compiler/xla/xla_data.proto b/tensorflow/compiler/xla/xla_data.proto
index 080e3c4267..06987e0044 100644
--- a/tensorflow/compiler/xla/xla_data.proto
+++ b/tensorflow/compiler/xla/xla_data.proto
@@ -433,6 +433,20 @@ message ConvolveRequest {
   ConvolutionDimensionNumbers dimension_numbers = 5;
 }
 
+enum FftType {
+  FFT = 0;    // Forward FFT; complex in, complex out.
+  IFFT = 1;   // Inverse FFT; complex in, complex out.
+  RFFT = 2;   // Forward real FFT; real in, fft_length / 2 + 1 complex out
+  IRFFT = 3;  // Inverse real FFT; fft_length / 2 + 1 complex in,
+              //                   fft_length real out
+}
+
+message FftRequest {
+  FftType fft_type = 1;
+  repeated int64 fft_length = 2;  // Multivalent for higher-order FFT.
+  ComputationDataHandle operand = 3;
+}
+
 message InfeedRequest {
   // The shape of the data returned by reading the device's infeed buffer.
   Shape shape = 2;
@@ -868,7 +882,8 @@ message OpRequest {
     BatchNormTrainingRequest batch_norm_training_request = 35;
     BatchNormGradRequest batch_norm_grad_request = 37;
     BatchNormInferenceRequest batch_norm_inference_request = 38;
-    // Next: 41
+    FftRequest fft_request = 41;
+    // Next: 42
   }
 }