diff options
Diffstat (limited to 'include/operator.def')
| -rw-r--r-- | include/operator.def | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/include/operator.def b/include/operator.def index 51eaf82..3c766c4 100644 --- a/include/operator.def +++ b/include/operator.def @@ -27,6 +27,7 @@ DEF_OPERATOR(AvgPool2d) DEF_OPERATOR(Conv2D) DEF_OPERATOR(Conv3D) DEF_OPERATOR(DepthwiseConv2D) +DEF_OPERATOR(FFT2d) DEF_OPERATOR(FullyConnected) DEF_OPERATOR(MatMul) DEF_OPERATOR(MaxPool2d) |