CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers
|
< Layout function
#include <tensor_fill.h>
Public Types | |
using | TensorView = TensorView< Element, Layout > |
Public Member Functions | |
TensorFillLinearFunc () | |
TensorFillLinearFunc (TensorView const &view_, Array< Element, Layout::kRank > const &v_, Element s_=Element(0)) | |
Constructs functor. More... | |
void | operator() (Coord< Layout::kRank > const &coord) const |
Updates the tensor. More... | |
Public Attributes | |
TensorView | view |
Array< Element, Layout::kRank > | v |
Element | s |
using cutlass::reference::host::detail::TensorFillLinearFunc< Element, Layout >::TensorView = TensorView<Element, Layout> |
|
inline |
|
inline |
|
inline |
Element cutlass::reference::host::detail::TensorFillLinearFunc< Element, Layout >::s |
Array<Element, Layout::kRank> cutlass::reference::host::detail::TensorFillLinearFunc< Element, Layout >::v |
TensorView cutlass::reference::host::detail::TensorFillLinearFunc< Element, Layout >::view |