declare explicitly the tblob default assign operator and copy constru…

…ctor (apache#15937)
anirudhacharya · Aug 20, 2019 · 47a0554 · 47a0554
1 parent 5bfd963
commit 47a0554
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/include/mxnet/tensor_blob.h b/include/mxnet/tensor_blob.h
@@ -148,6 +148,13 @@ class TBlob {
   TBlob(const mshadow::Tensor<Device, dim, DType> &src) {  // NOLINT(*)
     *this = src;
   }
+  /*!
+   * \brief constructor from TBlob (copy constructor)
+   * \param src source TBlob
+   */
+  TBlob(const TBlob &src): dptr_(src.dptr_), shape_(src.shape_), type_flag_(src.type_flag_) {
+    this->SetDLTensor(src.dev_mask(), src.dev_id());
+  }
   /*!
    * \brief assignment from tensor
    * \param src source tensor
@@ -164,6 +171,18 @@ class TBlob {
     SetDLTensor(Device::kDevMask, -1);
     return *this;
   }
+  /*!
+   * \brief assignment from TBlob (copy assignment)
+   * \param src source TBlob
+   * \return reference of self
+   */
+  inline TBlob &operator=(const TBlob &src) {
+    dptr_ = src.dptr_;
+    shape_ = src.shape_;
+    type_flag_ = src.type_flag_;
+    SetDLTensor(src.dev_mask(), src.dev_id());
+    return *this;
+  }
   /*!
    * \return whether the tensor's memory is continuous
    */