diff --git a/dnn/include/megdnn/basic_types.h b/dnn/include/megdnn/basic_types.h index 4c45755d5c83e12adebbf772ceae814b8aac7915..d49931c74d0fc2ba3007b3cf0662f9f30dd68c35 100644 --- a/dnn/include/megdnn/basic_types.h +++ b/dnn/include/megdnn/basic_types.h @@ -95,6 +95,7 @@ struct TensorShape { #if MEGDNN_CC_HOST TensorShape() = default; TensorShape(const TensorShape& rhs) = default; + MGE_WIN_DECLSPEC_FUC TensorShape& operator=(const TensorShape& rhs) = default; MGE_WIN_DECLSPEC_FUC TensorShape(const SmallVector& init_shape); MGE_WIN_DECLSPEC_FUC TensorShape(std::initializer_list init_shape); MGE_WIN_DECLSPEC_FUC std::string to_string() const; @@ -224,6 +225,7 @@ struct TensorLayout : public TensorShape { #if MEGDNN_CC_HOST TensorLayout(const TensorLayout& layout) = default; + TensorLayout& operator=(const TensorLayout& layout) = default; //! create empty layout with given dtype MGE_WIN_DECLSPEC_FUC explicit TensorLayout(DType dtype_);