Skip to content

Commit c64a736

Browse files
author
NeiroYT
committed
Fix argument mismatch
1 parent a75c39e commit c64a736

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

app/Graph/build.hpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -93,8 +93,8 @@ class LayerFactory {
9393

9494
static std::shared_ptr<Layer> createConvLayer(
9595
const RuntimeOptions& options, size_t step, size_t pads, size_t dilations,
96-
const Tensor& kernel, const Tensor& bias = Tensor(), size_t group = 1,
97-
bool useLegacyImpl = false) {
96+
Tensor& kernel, Tensor& bias = *std::make_shared<Tensor>(),
97+
size_t group = 1, bool useLegacyImpl = false) {
9898
if (options.backend == Backend::kOneDnn) {
9999
return std::make_shared<ConvLayerOneDnn>(step, pads, dilations, kernel,
100100
bias, group, useLegacyImpl);

0 commit comments

Comments
 (0)