Searched refs:InferFftShape (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.h | 118 static StatusOr<Shape> InferFftShape(const Shape& in, FftType fft_type,
|
| D | shape_inference_test.cc | 812 auto inferred_status = ShapeInference::InferFftShape(shape, type, length); in Pass()
820 auto inferred_status = ShapeInference::InferFftShape(shape, type, length); in Fail()
|
| D | hlo_verifier.cc | 164 ShapeInference::InferFftShape(fft->operand(0)->shape(), fft->fft_type(), in HandleFft()
|
| D | shape_inference.cc | 1873 /* static */ StatusOr<Shape> ShapeInference::InferFftShape( in InferFftShape() function in xla::ShapeInference
|
| D | hlo_parser.cc | 1670 return ShapeInference::InferFftShape(operands[0]->shape(), in ParseInstructionRhs()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1625 TF_ASSIGN_OR_RETURN(Shape shape, ShapeInference::InferFftShape( in Fft()
|