CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers
|
Parameters structure.
#include <tensor_fill.h>
Public Member Functions | |
CUTLASS_HOST_DEVICE | Params () |
Default ctor. More... | |
Params (TensorView view_=TensorView(), Element diag_=Element(1), Element other_=Element(0)) | |
Construction of Gaussian RNG functor. More... | |
Public Attributes | |
TensorView | view |
Element | diag |
Element | other |
|
inline |
|
inline |
Element cutlass::reference::device::detail::TensorFillDiagonalFunc< Element, Layout >::Params::diag |
Element cutlass::reference::device::detail::TensorFillDiagonalFunc< Element, Layout >::Params::other |
TensorView cutlass::reference::device::detail::TensorFillDiagonalFunc< Element, Layout >::Params::view |