Shape(std::initializer_list<int32_t> &&dims) : _dims(std::move(dims)) {}
- // TODO Make it 'explicit'.
- Shape(const std::vector<int32_t> &dims) : _dims(std::begin(dims), std::end(dims)) {}
+ explicit Shape(const std::vector<int32_t> &dims) : _dims(std::begin(dims), std::end(dims)) {}
int32_t rank() const { return static_cast<int32_t>(_dims.size()); }
.withType("Conv2D")
.withInShapes(getInputShapes(op))
.withOutShapes(getOutputShapes(op))
- .withStride(op.getStrides())
+ .withStride(Shape(op.getStrides()))
.withShape("Padding before", Shape(op.getPaddingBefore()))
.withShape("Padding after", Shape(op.getPaddingAfter()));
.withType("DepthwiseConv2D")
.withInShapes(getInputShapes(op))
.withOutShapes(getOutputShapes(op))
- .withStride(op.getStrides())
+ .withStride(Shape(op.getStrides()))
.withShape("Padding before", Shape(op.getPaddingBefore()))
.withShape("Padding after", Shape(op.getPaddingAfter()));
.withInShapes(getInputShapes(op))
.withOutShapes(getOutputShapes(op))
.withPadType(op.getPaddingType())
- .withStride(op.getStrides());
+ .withStride(Shape(op.getStrides()));
_dot_builder.updateWithOp(&op, node_info);
}