10 #include <arm_compute/core/Error.h> 11 #include <arm_compute/runtime/IFunction.h> 12 #include <arm_compute/runtime/CL/functions/CLSplit.h> 20 const std::vector<std::reference_wrapper<TensorInfo>>& outputs,
21 unsigned int splitAxis);
28 const arm_compute::CLCompileContext& clCompileContext);
33 std::unique_ptr<arm_compute::IFunction> m_Layer;
arm_compute::Status ClSplitterWorkloadValidate(const TensorInfo &input, const std::vector< std::reference_wrapper< TensorInfo >> &outputs, unsigned int splitAxis)
Copyright (c) 2021 ARM Limited and Contributors.
void Execute() const override
Contains information about inputs and outputs to a layer.
ClSplitterWorkload(const SplitterQueueDescriptor &descriptor, const WorkloadInfo &info, const arm_compute::CLCompileContext &clCompileContext)