24.05
|
virtual std::unique_ptr< IWorkload > CreateWorkload(const IWorkloadFactory &factory) const override
Makes a workload for the Softmax type.
void ExecuteStrategy(IStrategy &strategy) const override
Apply a visitor to this layer.
SoftmaxLayer * Clone(Graph &graph) const override
Creates a dynamically-allocated copy of this layer.
SoftmaxLayer(const SoftmaxDescriptor ¶m, const char *name)
Constructor to create a SoftmaxLayer.
~SoftmaxLayer()=default
Default destructor.
This layer represents a softmax operation.
Copyright (c) 2021 ARM Limited and Contributors.
void ValidateTensorShapesFromInputs() override
Check if the input tensor shape(s) will lead to a valid configuration of SoftmaxLayer.
A SoftmaxDescriptor for the SoftmaxLayer.