fixed a few small refactorings in EvalReader.h that I had missed
This commit is contained in:
Родитель
a03dd57e2a
Коммит
6878fa4b74
|
@ -175,13 +175,13 @@ public:
|
|||
void CopyMBLayoutTo(MBLayoutPtr pMBLayout)
|
||||
{
|
||||
assert(m_switchFrame.size() == 1);
|
||||
pMBLayout->Resize(1, m_mbSize);
|
||||
pMBLayout->Init(1, m_mbSize);
|
||||
|
||||
if (m_switchFrame[0] < m_mbSize) /* there is a switch frame within the minibatch*/
|
||||
{
|
||||
pMBLayout->Reset(0, m_switchFrame[0], MinibatchPackingFlags::SequenceStart);
|
||||
pMBLayout->Set(0, m_switchFrame[0], MinibatchPackingFlags::SequenceStart);
|
||||
if (m_switchFrame[0] > 0)
|
||||
pMBLayout->Reset(0, m_switchFrame[0] - 1, MinibatchPackingFlags::SequenceEnd);
|
||||
pMBLayout->SetWithoutOr(0, m_switchFrame[0] - 1, MinibatchPackingFlags::SequenceEnd); // TODO: can't we use Set()?
|
||||
}
|
||||
}
|
||||
|
||||
|
|
Загрузка…
Ссылка в новой задаче