ArmNN
NotReleased
|
#include "NeonSplitterWorkload.hpp"
#include <arm_compute/runtime/NEON/functions/NESplit.h>
#include <aclCommon/ArmComputeTensorUtils.hpp>
#include <aclCommon/ArmComputeUtils.hpp>
#include <backendsCommon/CpuTensorHandle.hpp>
#include <neon/NeonTensorHandle.hpp>
#include "NeonWorkloadUtils.hpp"
Go to the source code of this file.
Namespaces | |
armnn | |
Functions | |
arm_compute::Status | NeonSplitterWorkloadValidate (const TensorInfo &input, const std::vector< std::reference_wrapper< TensorInfo >> &outputs, unsigned int splitAxis) |