#include <RandomModel.h>

Collaboration diagram for nnsuite::conv::RandomModel:

Public Member Functions
	RandomModel (int32_t seed)

const nncc::core::ADT::feature::Shape &	ifm_shape (void) const override

const std::string &	ifm_name (void) const override

const nncc::core::ADT::feature::Shape &	ofm_shape (void) const override

const std::string &	ofm_name (void) const override

const nncc::core::ADT::kernel::Shape &	ker_shape (void) const override

const nncc::core::ADT::kernel::Reader< float > &	ker_data (void) const override

Public Member Functions inherited from nnsuite::conv::Model
virtual	~Model ()=default

Detailed Description

Definition at line 31 of file RandomModel.h.

Constructor & Destructor Documentation

◆ RandomModel()

nnsuite::conv::RandomModel::RandomModel ( int32_t seed )

explicit

Definition at line 30 of file RandomModel.cpp.

  : _ifm_shape{1, 8, 8}, _ifm_name{"ifm"}, _ofm_name{"ofm"}, _ofm_shape{2, 6, 6},
    _ker_buffer{kernel::Shape{2, 1, 3, 3}, kernel::NCHWLayout{}}
{
  std::default_random_engine gen{static_cast<uint32_t>(seed)};
  std::normal_distribution<float> dist{0.0f, 1.0f};
 
  const uint32_t N = _ker_buffer.shape().count();
  const uint32_t C = _ker_buffer.shape().depth();
  const uint32_t H = _ker_buffer.shape().height();
  const uint32_t W = _ker_buffer.shape().width();
 
  for (uint32_t n = 0; n < N; ++n)
  {
    for (uint32_t ch = 0; ch < C; ++ch)
    {
      for (uint32_t row = 0; row < H; ++row)
      {
        for (uint32_t col = 0; col < W; ++col)
        {
          _ker_buffer.at(n, ch, row, col) = dist(gen);
        }
      }
    }
  }
}