diff options
author | Michele Di Giorgio <michele.digiorgio@arm.com> | 2018-11-20 12:31:24 +0000 |
---|---|---|
committer | Michele Di Giorgio <michele.digiorgio@arm.com> | 2018-11-22 11:17:19 +0000 |
commit | 9d3a831d4131f8a8b37f127f11d36848d33e8496 (patch) | |
tree | 5ba1ab6ee69da474175ab82eda7ead31298af4c6 /src/core/CL/CLKernelLibrary.cpp | |
parent | 3112e33d8d0b987e85107390a0350bd5988f5f01 (diff) | |
download | ComputeLibrary-9d3a831d4131f8a8b37f127f11d36848d33e8496.tar.gz |
COMPMID-1648: CLNormalizationLayer IN_MAP_2D support for NHWC for FP32/FP16
Change-Id: I49f1d865f5e7562f1d80db849353a89ef77e6a9e
Diffstat (limited to 'src/core/CL/CLKernelLibrary.cpp')
-rw-r--r-- | src/core/CL/CLKernelLibrary.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/core/CL/CLKernelLibrary.cpp b/src/core/CL/CLKernelLibrary.cpp index 33e66705e3..3a002e808d 100644 --- a/src/core/CL/CLKernelLibrary.cpp +++ b/src/core/CL/CLKernelLibrary.cpp @@ -326,7 +326,8 @@ const std::map<std::string, std::string> CLKernelLibrary::_kernel_program_map = { "non_linear_filter_disk5x5", "non_linear_filter5x5.cl" }, { "non_max_suppression", "nonmax.cl" }, { "normalization_layer_cross_map", "normalization_layer.cl" }, - { "normalization_layer_in_map", "normalization_layer.cl" }, + { "normalization_layer_in_map_nchw", "normalization_layer.cl" }, + { "normalization_layer_in_map_nhwc", "normalization_layer.cl" }, { "normalize_planar_yuv_layer_nchw", "normalize_planar_yuv_layer.cl" }, { "normalize_planar_yuv_layer_nhwc", "normalize_planar_yuv_layer.cl" }, { "normalize_planar_yuv_layer_q8_nchw", "normalize_planar_yuv_layer_quantized.cl" }, |