diff options
| author | Adnan AlSinan <adnan.alsinan@arm.com> | 2021-09-01 15:32:03 +0100 |
|---|---|---|
| committer | Adnan AlSinan <adnan.alsinan@arm.com> | 2021-09-15 15:12:09 +0000 |
| commit | e4563a032aaa71de5efdb83fc04ff2933338e02d (patch) | |
| tree | f722dbea27753e0da68485df0592a128b72f747b /utils | |
| parent | e42a87ffd23a334b802b47a52ec28ad6c90bfbf0 (diff) | |
| download | ComputeLibrary-e4563a032aaa71de5efdb83fc04ff2933338e02d.tar.gz | |
Adds Conv3d reference implementation support.
Expands the interface with the following items:
- Size3D Class.
- Conv3dInfo Struct.
- Padding3D Struct.
- Add 'NDHWC' to supported Tensor Data Layouts.
- Add function to compute expected size of Conv3d.
Resolves COMPMID-4658 & COMPMID-4657
Signed-off-by: Adnan AlSinan <adnan.alsinan@arm.com>
Change-Id: Ic7452c48461eedaa38eaf3ac458f54b031e7dfa8
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6187
Reviewed-by: Giorgio Arena <giorgio.arena@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'utils')
| -rw-r--r-- | utils/TypePrinter.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/utils/TypePrinter.h b/utils/TypePrinter.h index 248c973b68..ee0135cce7 100644 --- a/utils/TypePrinter.h +++ b/utils/TypePrinter.h @@ -568,6 +568,9 @@ inline ::std::ostream &operator<<(::std::ostream &os, const DataLayout &data_lay case DataLayout::NCHW: os << "NCHW"; break; + case DataLayout::NDHWC: + os << "NDHWC"; + break; default: ARM_COMPUTE_ERROR("NOT_SUPPORTED!"); } |