10 #include <arm_compute/core/Error.h> 11 #include <arm_compute/runtime/IFunction.h> 19 const std::vector<std::reference_wrapper<TensorInfo>>& outputs,
20 unsigned int splitAxis);
30 mutable std::unique_ptr<arm_compute::IFunction> m_Layer;
NeonSplitterWorkload(const SplitterQueueDescriptor &descriptor, const WorkloadInfo &info)
void Execute() const override
arm_compute::Status NeonSplitterWorkloadValidate(const TensorInfo &input, const std::vector< std::reference_wrapper< TensorInfo >> &outputs, unsigned int splitAxis)