ContractionDesc< NumDTensor > Struct Template Reference#
ck::tensor_operation::device::ContractionDesc< NumDTensor > Struct Template Reference
#include <device_grouped_contraction_multiple_d.hpp>
Public Attributes | |
| std::vector< index_t > | a_ms_ks_lengths |
| std::vector< index_t > | a_ms_ks_strides |
| std::vector< index_t > | b_ns_ks_lengths |
| std::vector< index_t > | b_ns_ks_strides |
| std::array< std::vector< index_t >, NumDTensor > | ds_ms_ns_lengths |
| std::array< std::vector< index_t >, NumDTensor > | ds_ms_ns_strides |
| std::vector< index_t > | e_ms_ns_lengths |
| std::vector< index_t > | e_ms_ns_strides |
Member Data Documentation
◆ a_ms_ks_lengths
template<index_t NumDTensor>
| std::vector<index_t> ck::tensor_operation::device::ContractionDesc< NumDTensor >::a_ms_ks_lengths |
◆ a_ms_ks_strides
template<index_t NumDTensor>
| std::vector<index_t> ck::tensor_operation::device::ContractionDesc< NumDTensor >::a_ms_ks_strides |
◆ b_ns_ks_lengths
template<index_t NumDTensor>
| std::vector<index_t> ck::tensor_operation::device::ContractionDesc< NumDTensor >::b_ns_ks_lengths |
◆ b_ns_ks_strides
template<index_t NumDTensor>
| std::vector<index_t> ck::tensor_operation::device::ContractionDesc< NumDTensor >::b_ns_ks_strides |
◆ ds_ms_ns_lengths
template<index_t NumDTensor>
| std::array<std::vector<index_t>, NumDTensor> ck::tensor_operation::device::ContractionDesc< NumDTensor >::ds_ms_ns_lengths |
◆ ds_ms_ns_strides
template<index_t NumDTensor>
| std::array<std::vector<index_t>, NumDTensor> ck::tensor_operation::device::ContractionDesc< NumDTensor >::ds_ms_ns_strides |
◆ e_ms_ns_lengths
template<index_t NumDTensor>
| std::vector<index_t> ck::tensor_operation::device::ContractionDesc< NumDTensor >::e_ms_ns_lengths |
◆ e_ms_ns_strides
template<index_t NumDTensor>
| std::vector<index_t> ck::tensor_operation::device::ContractionDesc< NumDTensor >::e_ms_ns_strides |
The documentation for this struct was generated from the following file: