Searched refs:actLayer (Results 1 – 2 of 2) sorted by relevance
2048 …IConnectableLayer* actLayer = m_Network->AddActivationLayer(activationDescriptor, activationName.c… in ParseGemm() local2049 ARMNN_ASSERT(actLayer != nullptr); in ParseGemm()2051 … auto actInfo = ComputeOutputInfo({ activationName }, actLayer, { weightInfo.GetShape() }); in ParseGemm()2052 actLayer->GetOutputSlot(0).SetTensorInfo(actInfo[0]); in ParseGemm()2053 actLayer->GetOutputSlot(0).Connect(layer->GetInputSlot(1u)); in ParseGemm()2054 weightsLayer->GetOutputSlot(0).Connect(actLayer->GetInputSlot(0u)); in ParseGemm()2072 …IConnectableLayer* actLayer = m_Network->AddActivationLayer(activationDescriptor, activationName.c… in ParseGemm() local2073 ARMNN_ASSERT(actLayer != nullptr); in ParseGemm()2075 auto actInfo = ComputeOutputInfo({ activationName }, actLayer, { input1Shape }); in ParseGemm()2076 actLayer->GetOutputSlot(0).SetTensorInfo(actInfo[0]); in ParseGemm()[all …]
468 Layer* const actLayer = AddLayer<ActivationLayer>(activationDefaults, "act"); in InitialiseTestGraph() local469 actLayer->SetBackendId(armnn::Compute::CpuRef); in InitialiseTestGraph()471 concatLayer->GetOutputSlot(0).Connect(actLayer->GetInputSlot(0)); in InitialiseTestGraph()477 actLayer->GetOutputSlot(0).Connect(softmaxLayer->GetInputSlot(0)); in InitialiseTestGraph()491 actLayer->GetOutputSlot(0).SetEdgeStrategy(0, EdgeStrategy::DirectCompatibility); in InitialiseTestGraph()