ArmNN
 21.08
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 122 of file SerializeLayerParameters.hpp.

Member Function Documentation

◆ Serialize()

void Serialize ( ParameterStringifyFunction fn,
const ReduceDescriptor desc 
)
static

Definition at line 319 of file SerializeLayerParameters.cpp.

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

320 {
321  {
322  std::stringstream ss;
323  int count = 0;
324  for (auto&& var : desc.m_vAxis)
325  {
326  if (count > 0)
327  {
328  ss << ",";
329  }
330  ss << var;
331  ++count;
332  }
333  fn("Axis", ss.str());
334  }
335  fn("KeepDims", (desc.m_KeepDims ? "true" : "false"));
336  fn("ReduceOperation", GetReduceOperationAsCString(desc.m_ReduceOperation));
337 }
constexpr char const * GetReduceOperationAsCString(ReduceOperation reduce_operation)
Definition: TypesUtils.hpp:128

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