/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-6.4.3/include/ck/utility/reduction_enums.hpp Source File

/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-6.4.3/include/ck/utility/reduction_enums.hpp Source File#

Composable Kernel: /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-6.4.3/include/ck/utility/reduction_enums.hpp Source File
reduction_enums.hpp
Go to the documentation of this file.
1 // SPDX-License-Identifier: MIT
2 // Copyright (c) 2018-2023, Advanced Micro Devices, Inc. All rights reserved.
3 
4 #pragma once
5 
6 namespace ck {
7 
8 enum struct ReduceTensorOp
9 {
10  ADD = 0,
11  MUL = 1,
12  MIN = 2,
13  MAX = 3,
14  AMAX = 4,
15  AVG = 5,
16  NORM1 = 6,
17  NORM2 = 7,
18  // MUL_NO_ZEROS = 8,
19 };
20 
21 enum struct NanPropagation
22 {
23  NOT_PROPAGATE_NAN = 0,
24  PROPAGATE_NAN = 1,
25 };
26 
27 enum struct ReduceTensorIndices
28 {
29  NO_INDICES = 0,
30  FLATTENED_INDICES = 1,
31 };
32 
33 enum struct IndicesType
34 {
35  INDICES_32BIT = 0,
36  INDICES_64BIT = 1,
37  INDICES_16BIT = 2,
38  INDICES_8BIT = 3,
39 };
40 
41 } // namespace ck
ck
Definition: ck.hpp:264
ck::ReduceTensorOp
ReduceTensorOp
Definition: reduction_enums.hpp:9
ck::IndicesType
IndicesType
Definition: reduction_enums.hpp:34
ck::ReduceTensorIndices
ReduceTensorIndices
Definition: reduction_enums.hpp:28
ck::NanPropagation
NanPropagation
Definition: reduction_enums.hpp:22