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 |