[IE][VPU]: Enables check of parameters count for LSTMCell (#2335)
This commit is contained in:
parent
15ebc443d0
commit
171bf0fcc5
@ -19,7 +19,7 @@ set(VPU_SUPPORTED_FIRMWARES usb-ma2450 usb-ma2x8x pcie-ma248x)
|
||||
# Default packages
|
||||
#
|
||||
|
||||
set(FIRMWARE_PACKAGE_VERSION 1388)
|
||||
set(FIRMWARE_PACKAGE_VERSION 1390)
|
||||
set(VPU_CLC_MA2X8X_VERSION "movi-cltools-20.09.1")
|
||||
|
||||
#
|
||||
|
@ -83,6 +83,20 @@ private:
|
||||
bool useTempBuffer = (nCells > 1);
|
||||
IE_ASSERT((numTempBuffers() == 1 && useTempBuffer) || !useTempBuffer);
|
||||
|
||||
VPU_THROW_UNLESS(inputEdges().size() == 5,
|
||||
"LSTMCell: input edges: {}, but expected: 5",
|
||||
inputEdges().size());
|
||||
|
||||
// check number of outputs, without temp buffer
|
||||
const int outputsNumber = outputEdges().size();
|
||||
const int useCellState = outputsNumber >= 2;
|
||||
const int outputEdgesExpected = 1
|
||||
+ (useCellState ? 1 : 0)
|
||||
+ (outputsNumber == 3 ? 1 : 0);
|
||||
VPU_THROW_UNLESS(outputEdges().size() == outputEdgesExpected,
|
||||
"LSTMCell: number of output edges: {}, but expected: {}",
|
||||
outputEdges().size(), outputEdgesExpected);
|
||||
|
||||
for (const auto& inEdge : inputEdges()) {
|
||||
inEdge->input()->serializeBuffer(serializer);
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user