Changeset 1b9fe24 for src/ai

Timestamp:

Dec 29, 2021, 5:52:01 PM (3 years ago)

Author:

Paul Brossier <piem@piem.org>

Branches:

feature/crepe

Children:

ada7989

Parents:

0dad503

git-author:

Paul Brossier <piem@piem.org> (01/29/19 01:15:03)

git-committer:

Paul Brossier <piem@piem.org> (12/29/21 17:52:01)

Message:

[conv1d] add set_{kernel,bias}

File:

• src/ai/conv1d.c (modified) (1 diff)

src/ai/conv1d.c

@@ -340 +340 @@
 }
 
+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)
 {