ONE - On-device Neural Engine
Loading...
Searching...
No Matches
nnfw::cker::random::UniformFullIntDistribution32< Generator, IntType > Class Template Reference

#include <RandomDistributions.h>

Public Types

typedef Array< IntType, kResultElementCountResultType
 
typedef IntType ResultElementType
 

Public Member Functions

PHILOX_DEVICE_INLINE ResultType operator() (Generator *gen)
 

Static Public Attributes

static constexpr int kResultElementCount = Generator::kResultElementCount
 
static constexpr int kElementCost = 3
 
static constexpr bool kVariableSamplesPerOutput = false
 

Detailed Description

template<typename Generator, typename IntType>
class nnfw::cker::random::UniformFullIntDistribution32< Generator, IntType >

Definition at line 240 of file RandomDistributions.h.

Member Typedef Documentation

◆ ResultElementType

template<typename Generator , typename IntType >
typedef IntType nnfw::cker::random::UniformFullIntDistribution32< Generator, IntType >::ResultElementType

Definition at line 251 of file RandomDistributions.h.

◆ ResultType

template<typename Generator , typename IntType >
typedef Array<IntType, kResultElementCount> nnfw::cker::random::UniformFullIntDistribution32< Generator, IntType >::ResultType

Definition at line 250 of file RandomDistributions.h.

Member Function Documentation

◆ operator()()

template<typename Generator , typename IntType >
PHILOX_DEVICE_INLINE ResultType nnfw::cker::random::UniformFullIntDistribution32< Generator, IntType >::operator() ( Generator *  gen)
inline

Definition at line 254 of file RandomDistributions.h.

255 {
256 typename Generator::ResultType sample = (*gen)();
258 for (int i = 0; i < kResultElementCount; ++i)
259 {
260 result[i] = sample[i];
261 }
262 return result;
263 }
Array< IntType, kResultElementCount > ResultType
result
Definition infer.py:103

References nnfw::cker::random::UniformFullIntDistribution32< Generator, IntType >::kResultElementCount.

Field Documentation

◆ kElementCost

template<typename Generator , typename IntType >
constexpr int nnfw::cker::random::UniformFullIntDistribution32< Generator, IntType >::kElementCost = 3
staticconstexpr

Definition at line 246 of file RandomDistributions.h.

◆ kResultElementCount

template<typename Generator , typename IntType >
constexpr int nnfw::cker::random::UniformFullIntDistribution32< Generator, IntType >::kResultElementCount = Generator::kResultElementCount
staticconstexpr

◆ kVariableSamplesPerOutput

template<typename Generator , typename IntType >
constexpr bool nnfw::cker::random::UniformFullIntDistribution32< Generator, IntType >::kVariableSamplesPerOutput = false
staticconstexpr

Definition at line 249 of file RandomDistributions.h.


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