Commit 8b10c601 authored by Paul's avatar Paul
Browse files

Formatting

parent 2097978e
...@@ -18,9 +18,8 @@ struct check_shapes ...@@ -18,9 +18,8 @@ struct check_shapes
{ {
} }
template<class Op> template <class Op>
check_shapes(const shape* b, const shape* e, const Op& op) check_shapes(const shape* b, const shape* e, const Op& op) : begin(b), end(e), name(op.name())
: begin(b), end(e), name(op.name())
{ {
} }
......
...@@ -56,8 +56,9 @@ struct batch_norm_inference ...@@ -56,8 +56,9 @@ struct batch_norm_inference
shape compute_shape(std::vector<shape> inputs) const shape compute_shape(std::vector<shape> inputs) const
{ {
check_shapes{inputs, *this}.has(5); check_shapes{inputs, *this}.has(5);
check_shapes{inputs.data(), inputs.data()+1, *this}.only_dims(4); check_shapes{inputs.data(), inputs.data() + 1, *this}.only_dims(4);
check_shapes{inputs.data()+1, inputs.data()+inputs.size(), *this}.same_shape().elements(inputs.front().lens()[1]); check_shapes{inputs.data() + 1, inputs.data() + inputs.size(), *this}.same_shape().elements(
inputs.front().lens()[1]);
return inputs.front(); return inputs.front();
} }
}; };
......
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