From 3ff9654f52bbac06781cc240ba75e1a36ffe2ac9 Mon Sep 17 00:00:00 2001
From: Paul Brossier <piem@piem.org>
Date: Tue, 29 Jan 2019 01:15:03 +0100
Subject: [PATCH] [conv1d] add set_{kernel,bias}

---
 src/ai/conv1d.c | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/src/ai/conv1d.c b/src/ai/conv1d.c
index 34a71574..0bb419df 100644
--- a/src/ai/conv1d.c
+++ b/src/ai/conv1d.c
@@ -339,6 +339,26 @@ aubio_tensor_t *aubio_conv1d_get_kernel(aubio_conv1d_t* c)
   return c->kernel;
 }
 
+uint_t aubio_conv1d_set_kernel(aubio_conv1d_t *c, aubio_tensor_t *kernel)
+{
+  AUBIO_ASSERT(c && kernel);
+  if (aubio_tensor_have_same_shape(c->kernel, kernel)) {
+    aubio_tensor_copy(kernel, c->kernel);
+    return AUBIO_OK;
+  }
+  return AUBIO_FAIL;
+}
+
+uint_t aubio_conv1d_set_bias(aubio_conv1d_t *c, fvec_t *bias)
+{
+  AUBIO_ASSERT(c && bias);
+  if (bias->length == c->bias->length) {
+    fvec_copy(bias, c->bias);
+    return AUBIO_OK;
+  }
+  return AUBIO_FAIL;
+}
+
 fvec_t *aubio_conv1d_get_bias(aubio_conv1d_t* c)
 {
   AUBIO_ASSERT(c && c->bias);
-- 
2.11.0