Home
last modified time | relevance | path

Searched refs:reshapeOutputTensorShape (Results 1 – 2 of 2) sorted by relevance

/external/armnn/src/armnnDeserializer/
DDeserializer.cpp2075 const armnn::TensorShape& reshapeOutputTensorShape = reshapeOutputTensorInfo.GetShape(); in ParseReshape() local
2080 if (inputs.size() > 1 && !CheckShape(reshapeOutputTensorShape, expectedDims)) in ParseReshape()
2084 << reshapeOutputTensorShape in ParseReshape()
2093 reshapeDesc.m_TargetShape = reshapeOutputTensorShape; in ParseReshape()
/external/armnn/src/armnnTfLiteParser/
DTfLiteParser.cpp2108 const armnn::TensorShape& reshapeOutputTensorShape = reshapeOutputTensorInfo.GetShape(); in ParseReshape() local
2109 if (inputs.size() > 1 && !CheckShape(reshapeOutputTensorShape, outputs[0]->shape)) in ParseReshape()
2113 << reshapeOutputTensorShape in ParseReshape()