diff options
Diffstat (limited to 'include/armnn/backends/WorkloadInfo.hpp')
| -rw-r--r-- | include/armnn/backends/WorkloadInfo.hpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/include/armnn/backends/WorkloadInfo.hpp b/include/armnn/backends/WorkloadInfo.hpp index edf3581791..1d6967e134 100644 --- a/include/armnn/backends/WorkloadInfo.hpp +++ b/include/armnn/backends/WorkloadInfo.hpp @@ -11,12 +11,15 @@ namespace armnn { -/// Contains information about inputs and outputs to a layer. +/// Contains information about TensorInfos of a layer. /// This is needed at construction of workloads, but are not stored. struct WorkloadInfo { std::vector<TensorInfo> m_InputTensorInfos; std::vector<TensorInfo> m_OutputTensorInfos; + Optional<TensorInfo> m_WeightsTensorInfo = EmptyOptional(); + Optional<TensorInfo> m_BiasTensorInfo = EmptyOptional(); + Optional<std::string> m_ConvolutionMethod = EmptyOptional(); }; } //namespace armnn |