![]() |
CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers
|
Computes a random Gaussian distribution. More...
#include <tensor_fill.h>

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 | TensorUpdateOffDiagonalFunc (Params const ¶ms) |
| Device-side initialization of RNG. More... | |
| CUTLASS_DEVICE void | operator() (TensorCoord const &coord) |
| Compute random value and update RNG state. More... | |
Public Attributes | |
| Params | params |
| Parameters object. More... | |
< Layout function
| typedef TensorView::Element cutlass::reference::device::detail::TensorUpdateOffDiagonalFunc< Element, Layout >::T |
| typedef TensorView::TensorCoord cutlass::reference::device::detail::TensorUpdateOffDiagonalFunc< Element, Layout >::TensorCoord |
| using cutlass::reference::device::detail::TensorUpdateOffDiagonalFunc< Element, Layout >::TensorView = TensorView<Element, Layout> |
|
inline |
|
inline |
| Params cutlass::reference::device::detail::TensorUpdateOffDiagonalFunc< Element, Layout >::params |
1.8.11