xref: /aosp_15_r20/external/ComputeLibrary/src/core/AccessWindowAutoPadding.cpp (revision c217d954acce2dbc11938adb493fc0abd69584f3)
1 /*
2  * Copyright (c) 2017-2020 Arm Limited.
3  *
4  * SPDX-License-Identifier: MIT
5  *
6  * Permission is hereby granted, free of charge, to any person obtaining a copy
7  * of this software and associated documentation files (the "Software"), to
8  * deal in the Software without restriction, including without limitation the
9  * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
10  * sell copies of the Software, and to permit persons to whom the Software is
11  * furnished to do so, subject to the following conditions:
12  *
13  * The above copyright notice and this permission notice shall be included in all
14  * copies or substantial portions of the Software.
15  *
16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22  * SOFTWARE.
23  */
24 #include "src/core/AccessWindowAutoPadding.h"
25 
26 #include "arm_compute/core/ITensorInfo.h"
27 #include "arm_compute/core/Window.h"
28 
29 using namespace arm_compute;
30 
AccessWindowAutoPadding(ITensorInfo * info)31 AccessWindowAutoPadding::AccessWindowAutoPadding(ITensorInfo *info)
32     : _info(info)
33 {
34 }
35 
compute_valid_region(const Window & window,ValidRegion input_valid_region,bool border_undefined,BorderSize border_size) const36 ValidRegion AccessWindowAutoPadding::compute_valid_region(const Window &window, ValidRegion input_valid_region, bool border_undefined, BorderSize border_size) const
37 {
38     ARM_COMPUTE_UNUSED(window);
39     ARM_COMPUTE_UNUSED(input_valid_region);
40     ARM_COMPUTE_UNUSED(border_undefined);
41     ARM_COMPUTE_UNUSED(border_size);
42 
43     return compute_valid_region();
44 }
45 
compute_valid_region() const46 ValidRegion AccessWindowAutoPadding::compute_valid_region() const
47 {
48     if(_info == nullptr)
49     {
50         return ValidRegion{};
51     }
52 
53     return ValidRegion{ Coordinates(), _info->tensor_shape() };
54 }
55 
set_valid_region()56 void AccessWindowAutoPadding::set_valid_region()
57 {
58     if(_info == nullptr)
59     {
60         return;
61     }
62 
63     _info->set_valid_region(compute_valid_region());
64 }
65 
update_window_if_needed(Window & window) const66 bool AccessWindowAutoPadding::update_window_if_needed(Window &window) const
67 {
68     ARM_COMPUTE_UNUSED(window);
69 
70     return false;
71 }
72 
update_padding_if_needed(const Window & window)73 bool AccessWindowAutoPadding::update_padding_if_needed(const Window &window)
74 {
75     ARM_COMPUTE_UNUSED(window);
76 
77     // Only update the padding if the tensor allows it
78     if(_info == nullptr || !_info->is_resizable())
79     {
80         return false;
81     }
82 
83     // Update strides in tensor info
84     return _info->auto_padding();
85 }
86