Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 6 additions & 3 deletions NAM/dsp.h
Original file line number Diff line number Diff line change
Expand Up @@ -114,6 +114,7 @@ class DSP
virtual int PrewarmSamples() { return 0; };

virtual void SetMaxBufferSize(const int maxBufferSize);
int GetMaxBufferSize() const { return mMaxBufferSize; };

private:
struct Level
Expand Down Expand Up @@ -193,8 +194,7 @@ class Conv1D
long get_out_channels() const { return this->_weight.size() > 0 ? this->_weight[0].rows() : 0; };
int get_dilation() const { return this->_dilation; };

private:
// Gonna wing this...
protected:
// conv[kernel](cout, cin)
std::vector<Eigen::MatrixXf> _weight;
Eigen::VectorXf _bias;
Expand All @@ -217,10 +217,13 @@ class Conv1x1
void process_(const Eigen::MatrixXf& input, const int num_frames);

long get_out_channels() const { return this->_weight.rows(); };
long get_in_channels() const { return this->_weight.cols(); };

private:
protected:
Eigen::MatrixXf _weight;
Eigen::VectorXf _bias;

private:
Eigen::MatrixXf _output;
bool _do_bias;
};
Expand Down