diff options
Diffstat (limited to 'src/backends/reference/workloads/RefWorkloads.hpp')
| -rw-r--r-- | src/backends/reference/workloads/RefWorkloads.hpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/src/backends/reference/workloads/RefWorkloads.hpp b/src/backends/reference/workloads/RefWorkloads.hpp index 914137c23d..700a1d6184 100644 --- a/src/backends/reference/workloads/RefWorkloads.hpp +++ b/src/backends/reference/workloads/RefWorkloads.hpp @@ -14,6 +14,7 @@ #include "FullyConnected.hpp" #include "Gather.hpp" #include "Pooling2d.hpp" +#include "Pooling3d.hpp" #include "RefActivationWorkload.hpp" #include "RefArgMinMaxWorkload.hpp" #include "RefBatchNormalizationWorkload.hpp" @@ -51,6 +52,7 @@ #include "RefMeanWorkload.hpp" #include "RefNormalizationWorkload.hpp" #include "RefPooling2dWorkload.hpp" +#include "RefPooling3dWorkload.hpp" #include "RefPermuteWorkload.hpp" #include "RefPadWorkload.hpp" #include "RefPreluWorkload.hpp" |