Lines Matching refs:ComputeOutputInfo
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()
2021 auto transAInfo = ComputeOutputInfo({ transAName }, transALayer, { input0Shape }); in ParseGemm()
2051 … auto actInfo = ComputeOutputInfo({ activationName }, actLayer, { weightInfo.GetShape() }); in ParseGemm()
2075 auto actInfo = ComputeOutputInfo({ activationName }, actLayer, { input1Shape }); in ParseGemm()
2105 auto actInfo = ComputeOutputInfo({ activationName }, actLayer, { biasInfo.GetShape() }); in ParseGemm()
2136 auto actInfo = ComputeOutputInfo({ activationName }, in ParseGemm()
2150 auto outputInfo = ComputeOutputInfo({ node.output(0) }, layer, in ParseGemm()
2170 auto outputInfo = ComputeOutputInfo({node.output(0)}, layer, {inputShape}); in ParseGlobalAveragePool()
2198 …auto outputInfo = ComputeOutputInfo({node.output(0)}, layer, {inputShape}, onnx::TensorProto::INT6… in ParseShape()