Skip to content

Commit 89fb36a

Browse files
committed
Put lefCols() back in Dense layer Process calls
1 parent bb36717 commit 89fb36a

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

NeuralAudio/WaveNet.h

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -299,7 +299,7 @@ namespace NeuralAudio
299299
template<typename Derived, typename Derived2, typename Derived3>
300300
void Prewarm(const Eigen::MatrixBase<Derived>& layerInputs, const Eigen::MatrixBase<Derived2>& condition, Eigen::MatrixBase<Derived3> const& headInputs)
301301
{
302-
rechannel.Process(layerInputs, std::get<0>(layers).layerBuffer.middleCols(std::get<0>(layers).bufferStart, 1));
302+
rechannel.Process(layerInputs.leftCols(1), std::get<0>(layers).layerBuffer.middleCols(std::get<0>(layers).bufferStart, 1));
303303

304304
ForEachIndex<numLayers>([&](auto layerIndex)
305305
{
@@ -315,13 +315,13 @@ namespace NeuralAudio
315315
}
316316
});
317317

318-
headRechannel.Process(headInputs, headOutputs.leftCols(1));
318+
headRechannel.Process(headInputs.leftCols(1), headOutputs.leftCols(1));
319319
}
320320

321321
template<typename Derived, typename Derived2, typename Derived3>
322322
void Process(const Eigen::MatrixBase<Derived>& layerInputs, const Eigen::MatrixBase<Derived2>& condition, Eigen::MatrixBase<Derived3> const& headInputs, const size_t numFrames)
323323
{
324-
rechannel.Process(layerInputs, std::get<0>(layers).layerBuffer.middleCols(std::get<0>(layers).bufferStart, numFrames));
324+
rechannel.Process(layerInputs.leftCols(numFrames), std::get<0>(layers).layerBuffer.middleCols(std::get<0>(layers).bufferStart, numFrames));
325325

326326
ForEachIndex<numLayers>([&](auto layerIndex)
327327
{
@@ -337,7 +337,7 @@ namespace NeuralAudio
337337
std::get<layerIndex>(layers).AdvanceFrames(numFrames);
338338
});
339339

340-
headRechannel.Process(headInputs, headOutputs.leftCols(numFrames));
340+
headRechannel.Process(headInputs.leftCols(numFrames), headOutputs.leftCols(numFrames));
341341
}
342342
};
343343

0 commit comments

Comments
 (0)