diff options
| author | SiCongLi <sicong.li@arm.com> | 2021-02-24 15:40:57 +0000 |
|---|---|---|
| committer | SiCong Li <sicong.li@arm.com> | 2021-03-08 09:39:14 +0000 |
| commit | b88272e2dcf09803cd275cfeb5ca1ea743a091ff (patch) | |
| tree | 6637bef6d413bc513ffa451b0993102d7da57f4d /src/core/cpu/kernels/CpuSoftmaxKernel.cpp | |
| parent | c7b1e84ac5f3ada1b2f78c66979ef4d44804a955 (diff) | |
| download | ComputeLibrary-b88272e2dcf09803cd275cfeb5ca1ea743a091ff.tar.gz | |
Remove usage of valid window region in NHWC CPU kernels - Part2
Remove set_valid_region methods from all NHWC CPU ML functions /
operators / kernels
Resolves COMPMID-4152 (2/2)
Change-Id: If9725e9c5b0213b87db96675e81b7fb724970b98
Signed-off-by: SiCongLi <sicong.li@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5203
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/core/cpu/kernels/CpuSoftmaxKernel.cpp')
| -rw-r--r-- | src/core/cpu/kernels/CpuSoftmaxKernel.cpp | 10 |
1 files changed, 2 insertions, 8 deletions
diff --git a/src/core/cpu/kernels/CpuSoftmaxKernel.cpp b/src/core/cpu/kernels/CpuSoftmaxKernel.cpp index a8542b6be1..d2453ed21d 100644 --- a/src/core/cpu/kernels/CpuSoftmaxKernel.cpp +++ b/src/core/cpu/kernels/CpuSoftmaxKernel.cpp @@ -228,10 +228,7 @@ void CpuLogits1DMaxKernel::configure(const ITensorInfo *src, ITensorInfo *dst) // Output auto initialization if not yet initialized auto_init_if_empty(*dst, output_shape, 1, src->data_type(), src->quantization_info()); - Window win = calculate_max_window(*src, Steps()); - Coordinates coord; - coord.set_num_dimensions(dst->num_dimensions()); - dst->set_valid_region(ValidRegion(coord, dst->tensor_shape())); + Window win = calculate_max_window(*src, Steps()); ICpuKernel::configure(win); } @@ -330,10 +327,7 @@ void CpuLogits1DSoftmaxKernel<IS_LOG>::configure(const ITensorInfo *src, const I auto_init_if_empty(*tmp, TensorInfo(*src).set_data_type(tmp_data_type).reset_padding()); // Configure kernel window - Window win = calculate_max_window(*max, Steps()); - Coordinates coord; - coord.set_num_dimensions(dst->num_dimensions()); - dst->set_valid_region(ValidRegion(coord, dst->tensor_shape())); + Window win = calculate_max_window(*max, Steps()); ICpuKernel::configure(win); } |