keras::LayerDense Class Reference

#include "keras_model.h"

Inheritance diagram for keras::LayerDense:

Public Member Functions
	LayerDense ()
void	load_weights (std::ifstream &fin)
keras::DataChunk *	compute_output (keras::DataChunk *)
virtual unsigned int	get_input_rows () const
virtual unsigned int	get_input_cols () const
virtual unsigned int	get_output_units () const
std::string	get_name ()

Public Attributes
std::vector< std::vector< float > >	m_weights
std::vector< float >	m_bias
int	m_input_cnt
int	m_neurons
std::string	m_name

Detailed Description

Definition at line 212 of file keras_model.h.

Constructor & Destructor Documentation

keras::LayerDense::LayerDense ( )

inline

Definition at line 214 of file keras_model.h.

References fin.

: Layer("Dense") {}

Member Function Documentation

keras::DataChunk * keras::LayerDense::compute_output ( keras::DataChunk *  dc )

virtual

Implements keras::Layer.

Definition at line 381 of file keras_model.cc.

References keras::DataChunk::get_1d(), keras::DataChunkFlat::get_1d_rw(), util::size(), and w.

{
  //cout << "weights: input size " << m_weights.size() << endl;
  //cout << "weights: neurons size " << m_weights[0].size() << endl;
  //cout << "bias " << m_bias.size() << endl;
  size_t size = m_weights[0].size();
  size_t size8 = size >> 3;

  keras::DataChunkFlat* out = new DataChunkFlat(size, 0);
  float* y_ret = out->get_1d_rw().data();

  auto const& im = dc->get_1d();

  for (size_t j = 0; j < m_weights.size(); ++j) { // iter over input
    const float* w = m_weights[j].data();
    float p = im[j];
    size_t k = 0;
    for (size_t i = 0; i < size8; ++i) { // iter over neurons
      y_ret[k] += w[k] * p;              // vectorize if you can
      y_ret[k + 1] += w[k + 1] * p;
      y_ret[k + 2] += w[k + 2] * p;
      y_ret[k + 3] += w[k + 3] * p;
      y_ret[k + 4] += w[k + 4] * p;
      y_ret[k + 5] += w[k + 5] * p;
      y_ret[k + 6] += w[k + 6] * p;
      y_ret[k + 7] += w[k + 7] * p;
      k += 8;
    }
    while (k < size) {
      y_ret[k] += w[k] * p;
      ++k;
    }
  }
  for (size_t i = 0; i < size; ++i) { // add biases
    y_ret[i] += m_bias[i];
  }

  return out;
}

virtual unsigned int keras::LayerDense::get_input_cols ( ) const

inlinevirtual

Implements keras::Layer.

Definition at line 222 of file keras_model.h.

{ return m_input_cnt; }

virtual unsigned int keras::LayerDense::get_input_rows ( ) const

inlinevirtual

Implements keras::Layer.

Definition at line 221 of file keras_model.h.

{ return 1; } // flat, just one row

std::string keras::Layer::get_name ( )

inlineinherited

Definition at line 140 of file keras_model.h.

{ return m_name; }

virtual unsigned int keras::LayerDense::get_output_units ( ) const

inlinevirtual

Implements keras::Layer.

Definition at line 223 of file keras_model.h.

{ return m_neurons; }

void keras::LayerDense::load_weights ( std::ifstream &  fin )

virtual

Implements keras::Layer.

Definition at line 98 of file keras_model.cc.

References n.

{
  fin >> m_input_cnt >> m_neurons;
  float tmp_float;
  char tmp_char = ' ';
  for (int i = 0; i < m_input_cnt; ++i) {
    vector<float> tmp_n;
    fin >> tmp_char; // for '['
    for (int n = 0; n < m_neurons; ++n) {
      fin >> tmp_float;
      tmp_n.push_back(tmp_float);
    }
    fin >> tmp_char; // for ']'
    m_weights.push_back(tmp_n);
  }
  cout << "weights " << m_weights.size() << endl;
  fin >> tmp_char; // for '['
  for (int n = 0; n < m_neurons; ++n) {
    fin >> tmp_float;
    m_bias.push_back(tmp_float);
  }
  fin >> tmp_char; // for ']'
  cout << "bias " << m_bias.size() << endl;
}

Member Data Documentation

std::vector<float> keras::LayerDense::m_bias

Definition at line 219 of file keras_model.h.

int keras::LayerDense::m_input_cnt

Definition at line 225 of file keras_model.h.

std::string keras::Layer::m_name

inherited

Definition at line 141 of file keras_model.h.

int keras::LayerDense::m_neurons

Definition at line 226 of file keras_model.h.

std::vector<std::vector<float> > keras::LayerDense::m_weights

Definition at line 218 of file keras_model.h.

---

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

• larrecodnn/v09_23_00/source/larrecodnn/ImagePatternAlgs/Keras/keras_model.h
• larrecodnn/v09_23_00/source/larrecodnn/ImagePatternAlgs/Keras/keras_model.cc