diff options
| -rw-r--r-- | tensorflow/compiler/xla/xla_data.proto | 17 |
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 } } |