CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers
Classes | Namespaces | Functions
tensor_copy.h File Reference
#include <utility>
#include "cutlass/cutlass.h"
#include "tensor_foreach.h"
Include dependency graph for tensor_copy.h:

Go to the source code of this file.

Classes

struct  cutlass::reference::host::detail::TrivialConvert< DstElement, SrcElement >
 Helper to convert between types. More...
 
struct  cutlass::reference::host::detail::TensorCopyIf< DstElement, DstLayout, SrcElement, SrcLayout, F >
 Helper to conditionally copy between tensor views. More...
 

Namespaces

 cutlass
 
 cutlass::reference
 
 cutlass::reference::host
 
 cutlass::reference::host::detail
 Defines several helpers.
 

Functions

template<typename DstElement , typename DstLayout , typename SrcElement , typename SrcLayout , typename F >
void cutlass::reference::host::TensorCopy (TensorView< DstElement, DstLayout > dst, TensorView< SrcElement, SrcLayout > src, F const &transform)
 Copies elements from one tensor view into another, satisfying bounds of each tensor. More...
 
template<typename DstElement , typename DstLayout , typename SrcElement , typename SrcLayout , typename F >
void cutlass::reference::host::TensorCopy (TensorView< DstElement, DstLayout > dst, TensorRef< SrcElement, SrcLayout > src, F const &transform)
 
template<typename DstElement , typename DstLayout , typename SrcElement , typename SrcLayout , typename F >
void cutlass::reference::host::TensorCopy (TensorRef< DstElement, DstLayout > dst, TensorView< SrcElement, SrcLayout > src, F const &transform)
 
template<typename DstElement , typename DstLayout , typename SrcElement , typename SrcLayout >
void cutlass::reference::host::TensorCopy (TensorView< DstElement, DstLayout > dst, TensorView< SrcElement, SrcLayout > src)
 
template<typename DstElement , typename DstLayout , typename SrcElement , typename SrcLayout , typename F >
void cutlass::reference::host::TensorCopy (TensorView< DstElement, DstLayout > dst, TensorRef< SrcElement, SrcLayout > src)
 
template<typename DstElement , typename DstLayout , typename SrcElement , typename SrcLayout >
void cutlass::reference::host::TensorCopy (TensorRef< DstElement, DstLayout > dst, TensorView< SrcElement, SrcLayout > src)