convolution_host_tensor_descriptor_helper.hpp File Reference#
convolution_host_tensor_descriptor_helper.hpp File Reference
#include "ck/ck.hpp"#include "ck/tensor_operation/gpu/device/tensor_layout.hpp"#include "ck/library/utility/convolution_parameter.hpp"#include "ck/library/utility/host_tensor.hpp"Go to the source code of this file.
Namespaces | |
| namespace | ck |
| namespace | ck::utils |
| namespace | ck::utils::conv |
| namespace | ck::utils::conv::detail |
Functions | |
| template<typename OldLayout> | |
| std::vector< std::size_t > | ck::utils::conv::detail::get_layout_transpose_gnchw_to_old () |
| template<typename InLayout> | |
| HostTensorDescriptor | ck::utils::conv::make_input_host_tensor_descriptor_g_n_c_wis_packed (const ck::utils::conv::ConvParam ¶m) |
| template<typename WeiLayout> | |
| HostTensorDescriptor | ck::utils::conv::make_weight_host_tensor_descriptor_g_k_c_xs_packed (const ck::utils::conv::ConvParam ¶m) |
| template<typename OutLayout> | |
| HostTensorDescriptor | ck::utils::conv::make_output_host_tensor_descriptor_g_n_k_wos_packed (const ck::utils::conv::ConvParam ¶m) |