cutlass::reference::device::detail::TensorCopyDiagonalInFunc< Element, Layout > Struct Template Reference

Computes a random Gaussian distribution. More...

#include <tensor_fill.h>

Collaboration diagram for cutlass::reference::device::detail::TensorCopyDiagonalInFunc< Element, Layout >:

Classes
struct	Params
	Parameters structure. More...

Public Types
using	TensorView = TensorView< Element, Layout >
	View type. More...
typedef TensorView::Element	T
	Scalar type. More...
typedef TensorView::TensorCoord	TensorCoord
	Coordinate in tensor's index space. More...

Public Member Functions
CUTLASS_DEVICE	TensorCopyDiagonalInFunc (Params const &params)
	Device-side initialization of RNG. More...
CUTLASS_DEVICE void	operator() (TensorCoord const &coord)
	Only update the diagonal element. More...

Public Attributes
Params	params
	Parameters object. More...

Detailed Description

template<typename Element, typename Layout>
struct cutlass::reference::device::detail::TensorCopyDiagonalInFunc< Element, Layout >

< Layout function

Member Typedef Documentation

template<typename Element , typename Layout >

typedef TensorView::Element cutlass::reference::device::detail::TensorCopyDiagonalInFunc< Element, Layout >::T

template<typename Element , typename Layout >

typedef TensorView::TensorCoord cutlass::reference::device::detail::TensorCopyDiagonalInFunc< Element, Layout >::TensorCoord

template<typename Element , typename Layout >

using cutlass::reference::device::detail::TensorCopyDiagonalInFunc< Element, Layout >::TensorView = TensorView<Element, Layout>

Constructor & Destructor Documentation

template<typename Element , typename Layout >

CUTLASS_DEVICE cutlass::reference::device::detail::TensorCopyDiagonalInFunc< Element, Layout >::TensorCopyDiagonalInFunc ( Params const &  params )

inline

Member Function Documentation

template<typename Element , typename Layout >

CUTLASS_DEVICE void cutlass::reference::device::detail::TensorCopyDiagonalInFunc< Element, Layout >::operator() ( TensorCoord const &  coord )

inline

Member Data Documentation

template<typename Element , typename Layout >

Params cutlass::reference::device::detail::TensorCopyDiagonalInFunc< Element, Layout >::params

---

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

• device/tensor_fill.h