Searched refs:permuteLayer (Results 1 – 6 of 6) sorted by relevance
31 PermuteLayer* permuteLayer = graph.AddLayer<PermuteLayer>(desc, "permute"); variable33 permuteLayer->GetOutputSlot().SetTensorInfo(infoPermuted);38 constant->GetOutputSlot().Connect(permuteLayer->GetInputSlot(0));39 permuteLayer->GetOutputSlot().Connect(output->GetInputSlot(0));
62 armnn::Layer* const permuteLayer = GetFirstLayerWithName(graph, "permute_layer"); in PermuteDepthwiseConv2dWeightsTestRunner() local63 CHECK(permuteLayer); in PermuteDepthwiseConv2dWeightsTestRunner()70 TensorShape newShape = permuteLayer->GetOutputSlot().GetTensorInfo().GetShape(); in PermuteDepthwiseConv2dWeightsTestRunner()
60 PermuteLayer* permuteLayer = in AddPermuteLayer() local64 permuteLayer->GetOutputSlot().SetTensorInfo(weightsPermuted); in AddPermuteLayer()68 permuteLayer->SetBackendId(layer->GetBackendId()); in AddPermuteLayer()
93 PermuteLayer* permuteLayer) in ReplaceConstPermuteLayer() argument101 TensorInfo outputPermuteInfo = permuteLayer->GetOutputSlot(0).GetTensorInfo(); in ReplaceConstPermuteLayer()103 armnnUtils::Permute(outputPermuteInfo.GetShape(), permuteLayer->GetPermutation(), in ReplaceConstPermuteLayer()114 permuteLayer->GetOutputSlot().MoveAllConnections(constantLayer->GetOutputSlot()); in ReplaceConstPermuteLayer()
1107 IConnectableLayer* permuteLayer = m_Network->AddPermuteLayer(PermuteDescriptor(perVec), in AddConvLayerWithDepthwiseConv() local1109 permuteLayer->GetOutputSlot(0).SetTensorInfo(weightsPermuted); in AddConvLayerWithDepthwiseConv()1110 permuteLayer->GetOutputSlot(0).Connect(layer->GetInputSlot(1u)); in AddConvLayerWithDepthwiseConv()1113 weightsLayer->GetOutputSlot(0).Connect(permuteLayer->GetInputSlot(0u)); in AddConvLayerWithDepthwiseConv()
2093 …armnn::IConnectableLayer* const permuteLayer = network->AddPermuteLayer(descriptor, layerName.c_st… variable2096 inputLayer->GetOutputSlot(0).Connect(permuteLayer->GetInputSlot(0));2097 permuteLayer->GetOutputSlot(0).Connect(outputLayer->GetInputSlot(0));2100 permuteLayer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo);