CUTLASS: tensor_elementwise.h File Reference

| | CUTLASS

CUDA Templates for Linear Algebra Subroutines and Solvers |

• Main Page

• Modules

• Namespaces

• Classes

• Files

• File List

• File Members

• tools

• util

• include

• cutlass

• util

• reference

• device

• kernel

Namespaces | Functions

device/kernel/tensor_elementwise.h File Reference

#include <curand_kernel.h>
#include "cutlass/cutlass.h"

Include dependency graph for device/kernel/tensor_elementwise.h:

Go to the source code of this file.

|

Namespaces

| | | cutlass | | | | | cutlass::reference | | | | | cutlass::reference::device | | | | | cutlass::reference::device::kernel | | |

|

Functions

| | template<typename T > | | __global__ void | cutlass::reference::device::kernel::TensorInitializeUniform (Distribution dist, int64_t seed, int dim_contiguous, int dim_strided, T *tensor, int ldm) | | | Kernel to initialize tensor to uniform random distribution. More...
| | | | template<typename T > | | __global__ void | cutlass::reference::device::kernel::TensorInitializeGaussian (Distribution dist, int64_t seed, int dim_contiguous, int dim_strided, T *tensor, int ldm) | | | Kernel to initialize tensor to uniform distribution. More...
| | | | template<typename T > | | __global__ void | cutlass::reference::device::kernel::TensorInitializeLinear (Distribution dist, int64_t seed, int dim_contiguous, int dim_strided, T *tensor, int ldm) | | | Kernel to initialize tensor to an identity matrix. More...
| | | | template<typename T > | | __global__ void | cutlass::reference::device::kernel::TensorInitializeIdentity (Distribution dist, int64_t seed, int dim_contiguous, int dim_strided, T *tensor, int ldm) | | | Kernel to initialize tensor to an identity matrix. More...
| | |

---

Generated by 1.8.11