include/ck/tensor_operation/gpu/device/impl/device_contraction_utils.hpp File Reference

include/ck/tensor_operation/gpu/device/impl/device_contraction_utils.hpp File Reference#

Composable Kernel: include/ck/tensor_operation/gpu/device/impl/device_contraction_utils.hpp File Reference
device_contraction_utils.hpp File Reference
#include <cassert>
#include <sstream>
#include <vector>
#include "ck/ck.hpp"

Go to the source code of this file.

Namespaces

 ck
 
 ck::tensor_operation
 
 ck::tensor_operation::device
 

Functions

template<index_t NumDim1, index_t NumDim2>
auto ck::tensor_operation::device::CalculateMaxRead (const std::vector< index_t > &lengths, const std::vector< index_t > &strides)