Searched refs:rfft_op (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_tf.cc | 1361 auto rfft_op = dyn_cast<TF::RFFTOp>(op); in matchAndRewrite() local
1363 auto input = rfft_op.input(); in matchAndRewrite()
1366 auto fft_len = rfft_op.fft_length(); in matchAndRewrite()
1371 rfft_op.getResult().getType().dyn_cast_or_null<RankedTensorType>(); in matchAndRewrite()
1378 auto minus_two = CreateConstOpWithSingleValue(&rewriter, rfft_op.getLoc(), in matchAndRewrite()
1399 rfft_op.getLoc(), expaned_input_type, input, minus_two->getResult()); in matchAndRewrite()
1404 auto one = rewriter.create<TF::ConstOp>(rfft_op.getLoc(), one_attr); in matchAndRewrite()
1406 auto zero = CreateConstOpWithSingleValue(&rewriter, rfft_op.getLoc(), in matchAndRewrite()
1413 rfft_op.getLoc(), expanded_fft_len_type, in matchAndRewrite()
1420 rfft_op.getLoc(), rfft2d_out_type, expanded_input.getResult(), in matchAndRewrite()
[all …]
|