ArmNN
 21.11
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 132 of file SerializeLayerParameters.hpp.

Member Function Documentation

◆ Serialize()

void Serialize ( ParameterStringifyFunction fn,
const ReduceDescriptor desc 
)
static

Definition at line 354 of file SerializeLayerParameters.cpp.

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

355 {
356  {
357  std::stringstream ss;
358  int count = 0;
359  for (auto&& var : desc.m_vAxis)
360  {
361  if (count > 0)
362  {
363  ss << ",";
364  }
365  ss << var;
366  ++count;
367  }
368  fn("Axis", ss.str());
369  }
370  fn("KeepDims", (desc.m_KeepDims ? "true" : "false"));
371  fn("ReduceOperation", GetReduceOperationAsCString(desc.m_ReduceOperation));
372 }
constexpr char const * GetReduceOperationAsCString(ReduceOperation reduce_operation)
Definition: TypesUtils.hpp:139

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