ArmNN
 21.05
StringifyLayerParameters< ReduceDescriptor > Struct Template Reference

#include <SerializeLayerParameters.hpp>

Static Public Member Functions

static void Serialize (ParameterStringifyFunction &fn, const ReduceDescriptor &desc)
 

Detailed Description

template<>
struct armnn::StringifyLayerParameters< ReduceDescriptor >

Definition at line 63 of file SerializeLayerParameters.hpp.

Member Function Documentation

◆ Serialize()

void Serialize ( ParameterStringifyFunction fn,
const ReduceDescriptor desc 
)
static

Definition at line 138 of file SerializeLayerParameters.cpp.

References armnn::GetReduceOperationAsCString(), ReduceDescriptor::m_KeepDims, ReduceDescriptor::m_ReduceOperation, and ReduceDescriptor::m_vAxis.

139 {
140  {
141  std::stringstream ss;
142  int count = 0;
143  for (auto&& var : desc.m_vAxis)
144  {
145  if (count > 0)
146  {
147  ss << ",";
148  }
149  ss << var;
150  ++count;
151  }
152  fn("Axis", ss.str());
153  }
154  fn("KeepDims", (desc.m_KeepDims ? "true" : "false"));
155  fn("ReduceOperation", GetReduceOperationAsCString(desc.m_ReduceOperation));
156 }
constexpr char const * GetReduceOperationAsCString(ReduceOperation reduce_operation)
Definition: TypesUtils.hpp:126

The documentation for this struct was generated from the following files: