Changes

Author: NeiroYT

include/layers/ConvLayer.hpp

@@ -33,8 +33,8 @@ class ConvolutionalLayer : public Layer {
     pads_ = 0;
     dilations_ = 0;
   }
-  ConvolutionalLayer(size_t step, size_t pads, size_t dilations, Tensor& kernel,
-                     Tensor& bias = *std::make_shared<Tensor>(),
+  ConvolutionalLayer(size_t step, size_t pads, size_t dilations,
+                     const Tensor& kernel, const Tensor& bias = Tensor(),
                      size_t group = 1, bool useLegacyImpl = false)
       : Layer(kConvolution),
         kernel_(std::make_shared<Tensor>(kernel)),

include/layers/FCLayer.hpp

@@ -16,7 +16,7 @@ class FCLayer : public Layer {
 
  public:
   FCLayer() : Layer(kFullyConnected), weights_(nullptr), bias_(nullptr) {}
-  FCLayer(Tensor& weights, Tensor& bias)
+  FCLayer(const Tensor& weights, const Tensor& bias)
       : Layer(kFullyConnected),
         weights_(std::make_shared<Tensor>(weights)),
         bias_(std::make_shared<Tensor>(bias)) {}

include/layers_oneDNN/ConvLayer.hpp

@@ -21,9 +21,9 @@ class ConvLayerOneDnn : public Layer {
         group_(1),
         use_legacy_(false) {}
 
-  ConvLayerOneDnn(size_t stride, size_t pads, size_t dilations, Tensor& kernel,
-                  Tensor& bias = *std::make_shared<Tensor>(), size_t group = 1,
-                  bool use_legacy = false)
+  ConvLayerOneDnn(size_t stride, size_t pads, size_t dilations,
+                  const Tensor& kernel, const Tensor& bias = Tensor(),
+                  size_t group = 1, bool use_legacy = false)
       : Layer(kConvolution),
         stride_(stride),
         pads_(pads),