Searched refs:fft_op (Results 1 – 1 of 1) sorted by relevance
1570 auto fft_op = mlir::cast<mlir::lmhlo::FftOp>(input.op); in EmitFftThunkFromMlir() local1571 const Shape operand_shape = TypeToShape(fft_op.operand().getType()); in EmitFftThunkFromMlir()1572 const Shape output_shape = TypeToShape(fft_op.output().getType()); in EmitFftThunkFromMlir()1577 GetAllocationSliceForMlir(fft_op.operand())); in EmitFftThunkFromMlir()1579 GetAllocationSliceForMlir(fft_op.output())); in EmitFftThunkFromMlir()1580 TF_ASSIGN_OR_RETURN(xla::FftType fft_type, ConvertFftType(fft_op.fft_type())); in EmitFftThunkFromMlir()1581 auto fft_length_values = fft_op.fft_length().getValues<int64>(); in EmitFftThunkFromMlir()