diff options
Diffstat (limited to 'include/armnn/Types.hpp')
| -rw-r--r-- | include/armnn/Types.hpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/include/armnn/Types.hpp b/include/armnn/Types.hpp index 2fab6b44a9..ef52368365 100644 --- a/include/armnn/Types.hpp +++ b/include/armnn/Types.hpp @@ -53,7 +53,8 @@ enum class DataType enum class DataLayout { NCHW = 1, - NHWC = 2 + NHWC = 2, + NDHWC = 3 }; /// Define the behaviour of the internal profiler when outputting network details @@ -422,6 +423,8 @@ using InferenceTimingPair = std::pair<HighResolutionClock, HighResolutionClock>; X(Shape) \ X(UnidirectionalSequenceLstm) \ X(ChannelShuffle) \ + X(Convolution3d) \ + // New layers should be added at last to minimize instability. /// When adding a new layer, adapt also the LastLayer enum value in the |