30 const bool IsMemoryManaged =
true)
override;
44 std::vector<TensorShape>
InferOutputShapes(
const std::vector<TensorShape>& inputShapes)
const override;
58 template <
typename FactoryType>
void ValidateTensorShapesFromInputs() override
std::vector< TensorShape > InferOutputShapes(const std::vector< TensorShape > &inputShapes) const override
void Accept(ILayerVisitor &visitor) const override
virtual void CreateTensorHandles(const TensorHandleFactoryRegistry ®istry, const IWorkloadFactory &factory, const bool IsMemoryManaged=true) override
SplitterLayer * Clone(Graph &graph) const override
A ViewsDescriptor for the SplitterLayer. Descriptor to configure the splitting process. Number of Views must be equal to the number of outputs, and their order must match - e.g. first view corresponds to the first output, second view to the second output, etc.
virtual std::unique_ptr< IWorkload > CreateWorkload(const IWorkloadFactory &factory) const override
~SplitterLayer()=default
Default destructor.
This layer represents a split operation.
ClWorkloadFactory FactoryType
SplitterLayer(const ViewsDescriptor ¶m, const char *name)