24.02.1
cpp_nhwc_1x1_stride_any_depthfirst.hpp
Go to the documentation of this file.
1
/*
2
* Copyright (c) 2020-2022 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
25
#include <cstdint>
26
27
#pragma once
28
29
namespace
arm_conv
{
30
namespace
pooling
{
31
32
template
<
typename
T>
33
void
cpp_nhwc_1x1_stride_any_depthfirst_impl
(
const
uint64_t,
const
uint64_t, uint64_t n_channels,
const
T *
const
*
const
inptrs, T *outptr);
34
35
template
<
typename
T>
36
struct
cpp_nhwc_1x1_stride_any_depthfirst
:
IGenericDepthfirstStrategy
<T, T, Nothing>
37
{
38
using
Parent
=
IGenericDepthfirstStrategy<T, T, Nothing>
;
39
cpp_nhwc_1x1_stride_any_depthfirst
(
const
CPUInfo
*) {}
40
typename
Parent::KernelType
get_kernel
(
void
)
const override
{
return
cpp_nhwc_1x1_stride_any_depthfirst_impl<T>; }
41
};
42
43
}
// namespace pooling
44
}
// namespace arm_conv
arm_conv::pooling::cpp_nhwc_1x1_stride_any_depthfirst
Definition:
cpp_nhwc_1x1_stride_any_depthfirst.hpp:36
arm_conv::pooling::pooling
template UniquePoolingCommon< float, float > pooling(const PoolingArgs &, const Nothing &)
arm_conv::pooling::cpp_nhwc_1x1_stride_any_depthfirst_impl
void cpp_nhwc_1x1_stride_any_depthfirst_impl(uint64_t, uint64_t, const uint64_t n_channels, const T *const *const inptrs, T *outptr)
Definition:
generic.cpp:36
arm_compute::CPUInfo
Definition:
CPPTypes.h:66
arm_conv::pooling::cpp_nhwc_1x1_stride_any_depthfirst::cpp_nhwc_1x1_stride_any_depthfirst
cpp_nhwc_1x1_stride_any_depthfirst(const CPUInfo *)
Definition:
cpp_nhwc_1x1_stride_any_depthfirst.hpp:39
arm_conv::pooling::cpp_nhwc_1x1_stride_any_depthfirst::get_kernel
Parent::KernelType get_kernel(void) const override
Definition:
cpp_nhwc_1x1_stride_any_depthfirst.hpp:40
arm_compute::cpu::KernelType
kernels::CpuElementwiseUnaryKernel KernelType
Definition:
CpuElementwiseUnary.cpp:34
arm_conv::pooling::IGenericDepthfirstStrategy
Definition:
pooling_depthfirst_generic.hpp:37
arm_conv
Definition:
addressing.cpp:30
src
core
NEON
kernels
arm_conv
pooling
kernels
cpp_nhwc_1x1_stride_any_depthfirst.hpp
Generated on Mon Mar 18 2024 11:31:58 for Compute Library by
1.8.17