Home
last modified time | relevance | path

Searched refs:PermuteLayer (Results 1 – 25 of 37) sorted by relevance

12

/external/armnn/src/armnn/layers/
DPermuteLayer.cpp20 PermuteLayer::PermuteLayer(const PermuteDescriptor& param, const char* name) in PermuteLayer() function in armnn::PermuteLayer
25 std::unique_ptr<IWorkload> PermuteLayer::CreateWorkload(const IWorkloadFactory& factory) const in CreateWorkload()
33 PermuteLayer* PermuteLayer::Clone(Graph& graph) const in Clone()
35 return CloneBase<PermuteLayer>(graph, m_Param, GetName()); in Clone()
38 std::vector<TensorShape> PermuteLayer::InferOutputShapes(const std::vector<TensorShape>& inputShape… in InferOutputShapes()
45 void PermuteLayer::ValidateTensorShapesFromInputs() in ValidateTensorShapesFromInputs()
60 void PermuteLayer::ExecuteStrategy(IStrategy& strategy) const in ExecuteStrategy()
DPermuteLayer.hpp15 class PermuteLayer : public LayerWithParameters<PermuteDescriptor> class
25 PermuteLayer* Clone(Graph& graph) const override;
49 … GetPermutation().IsInverse(PolymorphicDowncast<const PermuteLayer*>(&other)->GetPermutation()); in IsInverse()
58 … GetPermutation().IsEqual(PolymorphicDowncast<const PermuteLayer*>(&other)->GetPermutation()); in IsEqual()
68 PermuteLayer(const PermuteDescriptor& param, const char* name);
71 ~PermuteLayer() = default;
/external/armnn/src/armnn/optimizations/
DConvertConstPermuteLayersToConstLayers.hpp35 … PolymorphicDowncast<PermuteLayer*>(&child)); in Run()
40 … PolymorphicDowncast<PermuteLayer*>(&child)); in Run()
45 … PolymorphicDowncast<PermuteLayer*>(&child)); in Run()
50 … PolymorphicDowncast<PermuteLayer*>(&child)); in Run()
55 … PolymorphicDowncast<PermuteLayer*>(&child)); in Run()
60 … PolymorphicDowncast<PermuteLayer*>(&child)); in Run()
65 … PolymorphicDowncast<PermuteLayer*>(&child)); in Run()
70 … PolymorphicDowncast<PermuteLayer*>(&child)); in Run()
75 … PolymorphicDowncast<PermuteLayer*>(&child)); in Run()
80 … PolymorphicDowncast<PermuteLayer*>(&child)); in Run()
[all …]
DMovePermuteUp.hpp33 auto permute = PolymorphicDowncast<PermuteLayer*>(&connection.GetOwningLayer()); in Run()
41PermuteLayer& permLayer = *graph.InsertNewLayer<PermuteLayer>(*baseInput, perm, name.c_str()); in Run()
83 using MovePermuteUp = OptimizeForConnection<Layer, PermuteLayer, MovePermuteUpImpl>;
DOptimizeInversePermutes.hpp42 using OptimizeInversePermutes = OptimizeForConnection<PermuteLayer, PermuteLayer,
43 OptimizeInversePermutesImpl<PermuteLayer>>;
DPermuteAsReshape.hpp18 void Run(Graph& graph, PermuteLayer& permute) const in Run()
39 static bool IsReshape(const PermuteLayer& layer) in IsReshape()
66 using PermuteAsReshape = OptimizeForType<PermuteLayer, PermuteAsReshapeImpl>;
DPermuteDepthwiseConv2dWeights.hpp60 PermuteLayer* permuteLayer = in AddPermuteLayer()
61 graph.InsertNewLayer<PermuteLayer>(layer->GetInputSlot(1), in AddPermuteLayer()
DSquashEqualSiblings.hpp67 …qualPermuteSiblings = OptimizeForConnection<Layer, PermuteLayer, SquashEqualSiblingsImpl<PermuteLa…
DPermuteAndBatchToSpaceAsDepthToSpace.hpp101 using PermuteAndBatchToSpaceAsDepthToSpace = OptimizeForConnection<PermuteLayer, BatchToSpaceNdLaye…
102 PermuteAndBatchToSpaceAsDepthToSpaceImpl<PermuteLayer>>;
/external/armnn/src/armnn/test/optimizations/
DMovePermuteUpTests.cpp30 …head = graph.InsertNewLayer<armnn::PermuteLayer>(head->GetInputSlot(0), armnn::PermuteDescriptor({…
75 … &IsLayerOfType<armnn::PermuteLayer>, &IsLayerOfType<armnn::OutputLayer>));
82 … &IsLayerOfType<armnn::PermuteLayer>, &IsLayerOfType<armnn::PermuteLayer>,
83 … &IsLayerOfType<armnn::PermuteLayer>, &IsLayerOfType<armnn::ElementwiseBinaryLayer>,
91 CHECK(CheckRelatedLayers<armnn::PermuteLayer>(graph, testRelatedLayers));
DSquashEqualSiblingsTests.cpp36 layer = graph.AddLayer<armnn::PermuteLayer>(permDesc, "");
53 layer = graph.AddLayer<armnn::PermuteLayer>(permDesc, "");
59 …aph.cbegin(), graph.cend(), &IsLayerOfType<armnn::InputLayer>, &IsLayerOfType<armnn::PermuteLayer>,
61 …&IsLayerOfType<armnn::PermuteLayer>, &IsLayerOfType<armnn::OutputLayer>, &IsLayerOfType<armnn::Out…
69 … &IsLayerOfType<armnn::PermuteLayer>, &IsLayerOfType<armnn::ReshapeLayer>,
DOptimizeInversePermutesTests.cpp27 …graph.InsertNewLayer<armnn::PermuteLayer>(output->GetInputSlot(0), armnn::PermuteDescriptor({ 0, 2…
29 …graph.InsertNewLayer<armnn::PermuteLayer>(output->GetInputSlot(0), armnn::PermuteDescriptor({ 0, 3…
33 … &IsLayerOfType<armnn::PermuteLayer>, &IsLayerOfType<armnn::PermuteLayer>,
DConvertConstPermuteLayersToConstLayersTest.cpp31 PermuteLayer* permuteLayer = graph.AddLayer<PermuteLayer>(desc, "permute");
43 &IsLayerOfType<PermuteLayer>,
DPermuteAsReshapeTests.cpp35 ….InsertNewLayer<armnn::PermuteLayer>(output->GetInputSlot(0), armnn::PermuteDescriptor({ 0, 2, 3, …
41 … &IsLayerOfType<armnn::PermuteLayer>, &IsLayerOfType<armnn::OutputLayer>));
DPermuteDepthwiseConv2dWeightsTests.cpp58 &IsLayerOfType<PermuteLayer>, in PermuteDepthwiseConv2dWeightsTestRunner()
DPermuteAndBatchToSpaceAsDepthToSpaceTests.cpp93 …eckSequence(graph.cbegin(), graph.cend(), &IsLayerOfType<InputLayer>, &IsLayerOfType<PermuteLayer>,
/external/armnn/src/armnnDeserializer/test/
DDeserializePermute.cpp50 layer_type: "PermuteLayer", in PermuteFixture()
54 layerName: "PermuteLayer", in PermuteFixture()
/external/ComputeLibrary/src/graph/nodes/
DPermuteLayerNode.cpp79 return NodeType::PermuteLayer; in type()
/external/ComputeLibrary/src/graph/backends/NEON/
DNENodeValidator.cpp96 case NodeType::PermuteLayer: in validate()
DNEFunctionFactory.cpp175 case NodeType::PermuteLayer: in create()
/external/ComputeLibrary/src/graph/backends/CL/
DCLNodeValidator.cpp99 case NodeType::PermuteLayer: in validate()
/external/ComputeLibrary/arm_compute/graph/
DTypePrinter.h143 case NodeType::PermuteLayer:
DTypes.h226 PermuteLayer, enumerator
/external/armnn/src/armnnSerializer/
DArmnnSchema.fbs585 table PermuteLayer {
1059 PermuteLayer,
/external/ComputeLibrary/arm_compute/graph/backends/
DValidateHelpers.h463 template <typename PermuteLayer>
475 return PermuteLayer::validate(input, output, perm); in validate_permute_layer()

12