/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck/tensor_operation/gpu/device/conv_tensor_rearrange_op.hpp Source File

• develop
• include
• ck
• tensor_operation
• gpu
• device

conv_tensor_rearrange_op.hpp

Go to the documentation of this file.

1 // SPDX-License-Identifier: MIT

2 // Copyright (c) 2023, Advanced Micro Devices, Inc. All rights reserved.

3 

4 #pragma once

5 

6 namespace ck {

7 namespace conv_tensor_rearrange_op {

8 

9 struct BaseConvTensorRearrangeOp

10 {

11 };

12 

13 struct ImageToColumn : public BaseConvTensorRearrangeOp

14 {

15  static constexpr const char* name = "Image to Column";

16 };

17 

18 struct ColumnToImage : public BaseConvTensorRearrangeOp

19 {

20  static constexpr const char* name = "Column to Image";

21 };

22 

23 template <typename Op,

24  typename std::enable_if<std::is_base_of<BaseConvTensorRearrangeOp, Op>::value,

25  bool>::type = false>

26 std::ostream& operator<<(std::ostream& os, const BaseConvTensorRearrangeOp&)

27 {

28  os << Op::name;

29  return os;

30 }

31 

32 } // namespace conv_tensor_rearrange_op

33 } // namespace ck

ck::conv_tensor_rearrange_op::operator<<

std::ostream & operator<<(std::ostream &os, const BaseConvTensorRearrangeOp &)

Definition: conv_tensor_rearrange_op.hpp:26

ck

Definition: ck.hpp:267

value

const GenericPointer< typename T::ValueType > T2 value

Definition: pointer.h:1350

ck::conv_tensor_rearrange_op::BaseConvTensorRearrangeOp

Definition: conv_tensor_rearrange_op.hpp:10

ck::conv_tensor_rearrange_op::ColumnToImage

Definition: conv_tensor_rearrange_op.hpp:19

ck::conv_tensor_rearrange_op::ColumnToImage::name

static constexpr const char * name

Definition: conv_tensor_rearrange_op.hpp:20

ck::conv_tensor_rearrange_op::ImageToColumn

Definition: conv_tensor_rearrange_op.hpp:14

ck::conv_tensor_rearrange_op::ImageToColumn::name

static constexpr const char * name

Definition: conv_tensor_rearrange_op.hpp:15