23.02
|
Go to the documentation of this file.
35 std::vector<TensorShape>
InferOutputShapes(
const std::vector<TensorShape>& inputShapes)
const override;
SliceLayer(const SliceDescriptor ¶m, const char *name)
Constructor to create a SliceLayer.
A SliceDescriptor for the SliceLayer.
virtual std::unique_ptr< IWorkload > CreateWorkload(const IWorkloadFactory &factory) const override
Makes a workload for the Slice type.
std::vector< TensorShape > InferOutputShapes(const std::vector< TensorShape > &inputShapes) const override
By default returns inputShapes if the number of inputs are equal to number of outputs,...
Copyright (c) 2021 ARM Limited and Contributors.
void ValidateTensorShapesFromInputs() override
Check if the input tensor shape(s) will lead to a valid configuration of SliceLayer.
~SliceLayer()=default
Default destructor.
void ExecuteStrategy(IStrategy &strategy) const override
Apply a visitor to this layer.
SliceLayer * Clone(Graph &graph) const override
Creates a dynamically-allocated copy of this layer.