diff options
Diffstat (limited to 'src/armnnTfLiteParser/TfLiteParser.cpp')
-rw-r--r-- | src/armnnTfLiteParser/TfLiteParser.cpp | 7 |
1 files changed, 2 insertions, 5 deletions
diff --git a/src/armnnTfLiteParser/TfLiteParser.cpp b/src/armnnTfLiteParser/TfLiteParser.cpp index 24fe7dad0b..e04f9adc9f 100644 --- a/src/armnnTfLiteParser/TfLiteParser.cpp +++ b/src/armnnTfLiteParser/TfLiteParser.cpp @@ -2276,13 +2276,10 @@ void TfLiteParser::ParseSplit(size_t subgraphIndex, size_t operatorIndex) auto inputTensorIndexes = AsUnsignedVector(GetInputTensorIds(m_Model, subgraphIndex, operatorIndex)); RegisterInputSlots(subgraphIndex, operatorIndex, layer, {inputTensorIndexes[1]}); - TensorShape outShape = TensorShape(static_cast<unsigned int>(splitterDimSizes.size()), - splitterDimSizes.data()); - for (unsigned int k = 0; k < layer->GetNumOutputSlots(); ++k) { - layer->GetOutputSlot(k).SetTensorInfo(armnn::TensorInfo(outShape, - inputTensorInfo.GetDataType())); + armnn::TensorInfo tensorInfo = ToTensorInfo(outputs[k]); + layer->GetOutputSlot(k).SetTensorInfo(tensorInfo); } auto outputTensorIndexes = AsUnsignedVector(GetOutputTensorIds(m_Model, subgraphIndex, operatorIndex)); |