nnfw::cker::random::SingleSampleAdapter< Generator > Class Template Reference

#include <RandomDistributions.h>

Public Types
typedef Generator::ResultElementType	ResultType
typedef Generator::ResultElementType	ResultElementType

Public Member Functions
PHILOX_DEVICE_INLINE	SingleSampleAdapter (Generator *gen)
PHILOX_DEVICE_INLINE ResultType	operator() ()
PHILOX_DEVICE_INLINE void	Skip (uint64_t num_skips)

Static Public Attributes
static constexpr int	kResultElementCount = 1
static constexpr int	kNativeElementCount = Generator::kResultElementCount

Detailed Description

template<class Generator>
class nnfw::cker::random::SingleSampleAdapter< Generator >

Definition at line 315 of file RandomDistributions.h.

Member Typedef Documentation

ResultElementType

template<class Generator >

typedef Generator::ResultElementType nnfw::cker::random::SingleSampleAdapter< Generator >::ResultElementType

Definition at line 323 of file RandomDistributions.h.

ResultType

template<class Generator >

typedef Generator::ResultElementType nnfw::cker::random::SingleSampleAdapter< Generator >::ResultType

Definition at line 322 of file RandomDistributions.h.

Constructor & Destructor Documentation

SingleSampleAdapter()

template<class Generator >

PHILOX_DEVICE_INLINE nnfw::cker::random::SingleSampleAdapter< Generator >::SingleSampleAdapter ( Generator *  gen )

inlineexplicit

Definition at line 326 of file RandomDistributions.h.

    : generator_(gen), used_result_index_(Generator::kResultElementCount)
  {
  }

Member Function Documentation

operator()()

template<class Generator >

PHILOX_DEVICE_INLINE ResultType nnfw::cker::random::SingleSampleAdapter< Generator >::operator() ( )

inline

Definition at line 332 of file RandomDistributions.h.

  {
    if (used_result_index_ == Generator::kResultElementCount)
    {
      unused_results_ = (*generator_)();
      used_result_index_ = 0;
    }
 
    return unused_results_[used_result_index_++];
  }

Skip()

template<class Generator >

PHILOX_DEVICE_INLINE void nnfw::cker::random::SingleSampleAdapter< Generator >::Skip ( uint64_t  num_skips )

inline

Definition at line 344 of file RandomDistributions.h.

  {
    if (!num_skips)
    {
      return;
    }
    int num_unused_results = kNativeElementCount - used_result_index_;
    if (num_skips <= num_unused_results)
    {
      used_result_index_ += num_skips;
      return;
    }
    num_skips -= num_unused_results;
    used_result_index_ = kNativeElementCount;
    SkipFromGenerator(num_skips / kNativeElementCount);
    num_skips = num_skips % kNativeElementCount;
    if (num_skips)
    {
      unused_results_ = (*generator_)();
      used_result_index_ = num_skips;
    }
  }

References nnfw::cker::random::SingleSampleAdapter< Generator >::kNativeElementCount.

Field Documentation

kNativeElementCount

template<class Generator >

constexpr int nnfw::cker::random::SingleSampleAdapter< Generator >::kNativeElementCount = Generator::kResultElementCount

staticconstexpr

Definition at line 321 of file RandomDistributions.h.

Referenced by nnfw::cker::random::SingleSampleAdapter< Generator >::Skip().

kResultElementCount

template<class Generator >

constexpr int nnfw::cker::random::SingleSampleAdapter< Generator >::kResultElementCount = 1

staticconstexpr

Definition at line 319 of file RandomDistributions.h.

---

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

• compute/cker/include/cker/operation/Helper/RandomDistributions.h