ArmNN
 22.02
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 137 of file SerializeLayerParameters.hpp.

Member Function Documentation

◆ Serialize()

void Serialize ( ParameterStringifyFunction fn,
const ReduceDescriptor desc 
)
static

Definition at line 383 of file SerializeLayerParameters.cpp.

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

384 {
385  {
386  std::stringstream ss;
387  int count = 0;
388  for (auto&& var : desc.m_vAxis)
389  {
390  if (count > 0)
391  {
392  ss << ",";
393  }
394  ss << var;
395  ++count;
396  }
397  fn("Axis", ss.str());
398  }
399  fn("KeepDims", (desc.m_KeepDims ? "true" : "false"));
400  fn("ReduceOperation", GetReduceOperationAsCString(desc.m_ReduceOperation));
401 }
constexpr char const * GetReduceOperationAsCString(ReduceOperation reduce_operation)
Definition: TypesUtils.hpp:139

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