conv_tensor_rearrange_op.hpp Source File

conv_tensor_rearrange_op.hpp Source File#

Composable Kernel: conv_tensor_rearrange_op.hpp Source File

// SPDX-License-Identifier: MIT
// Copyright (c) 2023, Advanced Micro Devices, Inc. All rights reserved.
 
#pragma once
 
namespace ck {
namespace conv_tensor_rearrange_op {
 
struct BaseConvTensorRearrangeOp
{
};
 
struct ImageToColumn : public BaseConvTensorRearrangeOp
{
    static constexpr const char* name = "Image to Column";
};
 
struct ColumnToImage : public BaseConvTensorRearrangeOp
{
    static constexpr const char* name = "Column to Image";
};
 
template <typename Op,
          typename std::enable_if<std::is_base_of<BaseConvTensorRearrangeOp, Op>::value,
                                  bool>::type = false>
std::ostream& operator<<(std::ostream& os, const BaseConvTensorRearrangeOp&)
{
    os << Op::name;
    return os;
}
 
} // namespace conv_tensor_rearrange_op
} // namespace ck