src/backends/WorkloadInfo.hpp

   1 //
   2 // Copyright © 2017 Arm Ltd. All rights reserved.
   3 // SPDX-License-Identifier: MIT
   4 //
   5 #pragma once
   6
   7 namespace armnn
   8 {
   9
  10 /// Contains information about inputs and outputs to a layer.
  11 /// This is needed at construction of workloads, but are not stored.
  12 struct WorkloadInfo
  13 {
  14     std::vector<TensorInfo> m_InputTensorInfos;
  15     std::vector<TensorInfo> m_OutputTensorInfos;
  16 };
  17
  18 } //namespace armnn