SMAUG
Simulating Machine Learning Applications on gem5-Aladdin
|
1 #include "smaug/core/operator.h"
6 bool anyInputDead =
false;
16 void Operator::printSummary(std::ostream& out)
const {
17 boost::format fmter(kLayerFormat);
18 out << fmter % (this->name +
" (" + OpType_Name(opType) +
")") %
21 for (
int i = 1; i <
outputs.size(); i++)
22 out << fmter %
"" %
outputs.at(i)->getShape() %
"";
virtual int getNumParameters() const
This returns the number of parameterizable weights in the operator.
std::vector< TensorBase * > outputs
An ordered list of output tensors produced by this operator.
The smaug namespace is the parent namespace of all C++ code in SMAUG.
std::vector< TensorBase * > inputs
An ordered list of input tensors consumed by this operator.
virtual bool isDead()
Returns true if the Operator is dead.