Wrapper around gen_spectral_ops.rfft* that infers fft_length argument.
(input_tensor, fft_length=None, name=None)
| 112 | """Wrapper around gen_spectral_ops.rfft* that infers fft_length argument.""" |
| 113 | |
| 114 | def _rfft(input_tensor, fft_length=None, name=None): |
| 115 | """Wrapper around gen_spectral_ops.rfft* that infers fft_length argument.""" |
| 116 | with _ops.name_scope(name, default_name, |
| 117 | [input_tensor, fft_length]) as name: |
| 118 | input_tensor = _ops.convert_to_tensor(input_tensor, _dtypes.float32) |
| 119 | input_tensor.shape.with_rank_at_least(fft_rank) |
| 120 | if fft_length is None: |
| 121 | fft_length = _infer_fft_length_for_rfft(input_tensor, fft_rank) |
| 122 | else: |
| 123 | fft_length = _ops.convert_to_tensor(fft_length, _dtypes.int32) |
| 124 | input_tensor = _maybe_pad_for_rfft(input_tensor, fft_rank, fft_length) |
| 125 | return fft_fn(input_tensor, fft_length, name) |
| 126 | _rfft.__doc__ = fft_fn.__doc__ |
| 127 | return _rfft |
| 128 |
nothing calls this directly
no test coverage detected