cutlass::reference::host::detail::TensorFillDiagonalFunc< Element, Layout > Struct Template Reference

< Layout function

#include <tensor_fill.h>

Collaboration diagram for cutlass::reference::host::detail::TensorFillDiagonalFunc< Element, Layout >:

Public Types
using	TensorView = TensorView< Element, Layout >

Public Member Functions
	TensorFillDiagonalFunc (TensorView const &view_=TensorView(), Element diag_=Element(1), Element other_=Element(0))
void	operator() (Coord< Layout::kRank > const &coord) const

Public Attributes
TensorView	view
Element	diag
Element	other

Member Typedef Documentation

template<typename Element, typename Layout>

using cutlass::reference::host::detail::TensorFillDiagonalFunc< Element, Layout >::TensorView = TensorView<Element, Layout>

Constructor & Destructor Documentation

template<typename Element, typename Layout>

cutlass::reference::host::detail::TensorFillDiagonalFunc< Element, Layout >::TensorFillDiagonalFunc ( TensorView const &  view_ = TensorView(), Element  diag_ = Element(1), Element  other_ = Element(0)  )

inline

Member Function Documentation

template<typename Element, typename Layout>

void cutlass::reference::host::detail::TensorFillDiagonalFunc< Element, Layout >::operator() ( Coord< Layout::kRank > const &  coord ) const

inline

Member Data Documentation

template<typename Element, typename Layout>

Element cutlass::reference::host::detail::TensorFillDiagonalFunc< Element, Layout >::diag

template<typename Element, typename Layout>

Element cutlass::reference::host::detail::TensorFillDiagonalFunc< Element, Layout >::other

template<typename Element, typename Layout>

TensorView cutlass::reference::host::detail::TensorFillDiagonalFunc< Element, Layout >::view

---

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

• host/tensor_fill.h