Home
last modified time | relevance | path

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

/external/armnn/src/armnnOnnxParser/
DOnnxParser.cpp2019 …IConnectableLayer* transALayer = m_Network->AddTransposeLayer(transposeADescriptor, transAName.c_s… in ParseGemm() local
2020 ARMNN_ASSERT(transALayer != nullptr); in ParseGemm()
2021 auto transAInfo = ComputeOutputInfo({ transAName }, transALayer, { input0Shape }); in ParseGemm()
2022 transALayer->GetOutputSlot(0).SetTensorInfo(transAInfo[0]); in ParseGemm()
2023 transALayer->GetOutputSlot(0).Connect(layer->GetInputSlot(0u)); in ParseGemm()
2025 RegisterInputSlot(transALayer, node.input(0), 0); in ParseGemm()