Commit 45665c93 authored by Khalique's avatar Khalique
Browse files

added reflect() for lrn

parent dbcbaeeb
...@@ -68,17 +68,19 @@ struct lrn ...@@ -68,17 +68,19 @@ struct lrn
int size = 1; int size = 1;
std::string name() const { return "lrn"; } std::string name() const { return "lrn"; }
template <class Self, class F>
static auto reflect(Self& self, F f)
{
return pack(
f(self.alpha, "alpha"), f(self.beta, "beta"), f(self.bias, "bias"), f(self.size, "size"));
}
shape compute_shape(std::vector<shape> inputs) const shape compute_shape(std::vector<shape> inputs) const
{ {
check_shapes{inputs, *this}.has(1); check_shapes{inputs, *this}.has(1);
return inputs.front(); return inputs.front();
} }
friend std::ostream& operator<<(std::ostream& os, const lrn& op)
{
os << op.name() << ":" << op.alpha << ":" << op.beta << ":" << op.bias << ":" << op.size;
return os;
}
}; };
struct convolution struct convolution
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment