CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers
Public Member Functions | Public Attributes | List of all members
cutlass::reference::device::detail::TensorFillDiagonalFunc< Element, Layout >::Params Struct Reference

Parameters structure.

#include <tensor_fill.h>

Collaboration diagram for cutlass::reference::device::detail::TensorFillDiagonalFunc< Element, Layout >::Params:
Collaboration graph
[legend]

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
 

Constructor & Destructor Documentation

template<typename Element , typename Layout >
CUTLASS_HOST_DEVICE cutlass::reference::device::detail::TensorFillDiagonalFunc< Element, Layout >::Params::Params ( )
inline
template<typename Element , typename Layout >
cutlass::reference::device::detail::TensorFillDiagonalFunc< Element, Layout >::Params::Params ( TensorView  view_ = TensorView(),
Element  diag_ = Element(1),
Element  other_ = Element(0) 
)
inline

Member Data Documentation

template<typename Element , typename Layout >
Element cutlass::reference::device::detail::TensorFillDiagonalFunc< Element, Layout >::Params::diag
template<typename Element , typename Layout >
Element cutlass::reference::device::detail::TensorFillDiagonalFunc< Element, Layout >::Params::other
template<typename Element , typename Layout >
TensorView cutlass::reference::device::detail::TensorFillDiagonalFunc< Element, Layout >::Params::view

The documentation for this struct was generated from the following file: