diff options
author | 2018-01-02 13:01:59 -0800 | |
---|---|---|
committer | 2018-01-02 13:05:54 -0800 | |
commit | 5bf26acd87d3d44183fc28cb9576cda10c0255ca (patch) | |
tree | 5486c0ab9077496e30b594d35c089a45f7b7e18c /tensorflow/compiler/xla/service/shape_inference.h | |
parent | 97a843db78745fe3a8e418b3b1e93ef79fbfff12 (diff) |
Automated g4 rollback of changelist 180000981
PiperOrigin-RevId: 180581912
Diffstat (limited to 'tensorflow/compiler/xla/service/shape_inference.h')
-rw-r--r-- | tensorflow/compiler/xla/service/shape_inference.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/shape_inference.h b/tensorflow/compiler/xla/service/shape_inference.h index c06340d2d5..b39151ebbc 100644 --- a/tensorflow/compiler/xla/service/shape_inference.h +++ b/tensorflow/compiler/xla/service/shape_inference.h @@ -109,6 +109,11 @@ class ShapeInference { const Shape& lhs, const Shape& rhs, const Window& window, const ConvolutionDimensionNumbers& dimension_numbers); + // Infers the shape produced by the given FFT type on the given operand. + static StatusOr<Shape> InferFftShape( + const Shape& in, FftType fft_type, + tensorflow::gtl::ArraySlice<int64> fft_length); + // Infers the shape produced a cross replica sum with the given operand // shapes. static StatusOr<Shape> InferCrossReplicaSumShape( |