34 std::vector<TensorShape>
InferOutputShapes(
const std::vector<TensorShape>& inputShapes)
const override;
bool IsEqual(const Layer &other) const
LayerType GetType() const
bool IsEqual(const PermutationVector &other) const
const PermutationVector & GetPermutation() const
bool IsInverse(const PermutationVector &other) const
PermuteLayer * Clone(Graph &graph) const override
void ValidateTensorShapesFromInputs() override
virtual std::unique_ptr< IWorkload > CreateWorkload(const IWorkloadFactory &factory) const override
This layer represents a permutation operation.
PermuteDescriptor m_Param
The parameters for the layer (not including tensor-valued weights etc.).
void Accept(ILayerVisitor &visitor) const override
bool IsInverse(const Layer &other) const
std::vector< TensorShape > InferOutputShapes(const std::vector< TensorShape > &inputShapes) const override
PermutationVector m_DimMappings
Indicates how to translate tensor elements from a given source into the target destination, when source and target potentially have different memory layouts e.g. {0U, 3U, 1U, 2U}.
~PermuteLayer()=default
Default destructor.
A PermuteDescriptor for the PermuteLayer.
PermuteLayer(const PermuteDescriptor ¶m, const char *name)