We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 033fa65 commit c70208bCopy full SHA for c70208b
tf2onnx/onnx_opset/signal.py
@@ -101,7 +101,10 @@ def DFT_real(x, fft_length=None):
101
"""
102
consumers = ctx.find_output_consumers(node.output[0])
103
consumer_types = set(op.type for op in consumers)
104
- utils.make_sure(consumer_types == {'ComplexAbs'}, "Current implementation of RFFT only allows ComplexAbs as consumer not %r", consumer_types)
+ utils.make_sure(
105
+ consumer_types == {'ComplexAbs'},
106
+ "Current implementation of RFFT only allows ComplexAbs as consumer not %r",
107
+ consumer_types)
108
109
onnx_dtype = ctx.get_dtype(node.input[0])
110
shape = ctx.get_shape(node.input[0])
0 commit comments