Changeset 3ff9654

Timestamp:

Jan 29, 2019, 1:15:03 AM (5 years ago)

Author:

Paul Brossier <piem@piem.org>

Branches:

feature/crepe_org

Children:

e2010b3

Parents:

cc74a29

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)
 {