ConcatSpec Class Reference

#include <ConcatSpec.h>

Public Member Functions
	ConcatSpec (uint32_t axis)
nncc::core::ADT::tensor::Shape	forward (const ShapeList &) const
	Return the output shape when inputs of given shape are concatenated along _axis.

Detailed Description

Definition at line 26 of file ConcatSpec.h.

Constructor & Destructor Documentation

ConcatSpec()

ConcatSpec::ConcatSpec ( uint32_t  axis )

inlineexplicit

Definition at line 29 of file ConcatSpec.h.

                                     : _axis{axis}
  {
    // DO NOTHING
  }

Member Function Documentation

forward()

nncc::core::ADT::tensor::Shape ConcatSpec::forward

(

const ShapeList &

inputs

)

const

Return the output shape when inputs of given shape are concatenated along _axis.

Definition at line 23 of file ConcatSpec.cpp.

{
  assert(inputs.size() > 0);
 
  Shape output_shape = inputs.at(0);
 
  for (uint32_t n = 1; n < inputs.size(); ++n)
  {
    // The current implementation assumes that "inputs" is well-formed
    // TODO Verify whether "inputs" is really well-formed
    const auto &input_shape = inputs.at(n);
    output_shape.dim(_axis) += input_shape.dim(_axis);
  }
 
  return output_shape;
}

References output_shape.

---

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

• compiler/enco/frontend/caffe/src/ConcatSpec.h
• compiler/enco/frontend/caffe/src/ConcatSpec.cpp