Searched refs:ComputeOutputInfo (Results 1 – 2 of 2) sorted by relevance
| /external/armnn/src/armnnOnnxParser/ |
| D | OnnxParser.cpp | 498 std::vector<TensorInfo> OnnxParserImpl::ComputeOutputInfo(std::vector<std::string> outNames, in ComputeOutputInfo() function in armnnOnnxParser::OnnxParserImpl
1136 auto outputInfo = ComputeOutputInfo({ node.output(0) }, layer, in AddConvLayerWithDepthwiseConv()
1248 auto outputInfo = ComputeOutputInfo({ outputName }, in AddFullyConnected()
1391 …auto outputInfo = ComputeOutputInfo({node.output(0)}, layer, {m_TensorsInfo[node.input(0)].m_info-… in AddPoolingLayer()
1488 …auto outputInfo = ComputeOutputInfo({ node.output(0)}, layer, {m_TensorsInfo[node.input(0)].m_info… in ParseActivation()
1564 auto outputInfo = ComputeOutputInfo({ node.output(0) }, layer, in ParseAdd()
1633 …auto outputInfo = ComputeOutputInfo({node.output(0)}, layer, {m_TensorsInfo[node.input(0)].m_info-… in ParseBatchNormalization()
1677 auto outputInfo = ComputeOutputInfo({node.output(0)}, layer, inputShapes, in ParseConcat()
1904 auto outputInfo = ComputeOutputInfo({ node.output(0) }, layer, in ParseConv()
1977 auto outputInfo = ComputeOutputInfo({node.output(0)}, layer, { inputShape, indicesShape }, in ParseGather()
[all …]
|
| D | OnnxParser.hpp | 98 std::vector<armnn::TensorInfo> ComputeOutputInfo(
|