Home
last modified time | relevance | path

Searched refs:InferFftShape (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.h116 static StatusOr<Shape> InferFftShape(const Shape& in, FftType fft_type,
Dhlo_verifier.cc167 ShapeInference::InferFftShape(fft->operand(0)->shape(), fft->fft_type(), in HandleFft()
Dshape_inference.cc1818 /* static */ StatusOr<Shape> ShapeInference::InferFftShape( in InferFftShape() function in xla::ShapeInference
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1269 TF_ASSIGN_OR_RETURN(Shape shape, ShapeInference::InferFftShape( in Fft()