change input[1] of embed as parameter

sony · May 10, 2018 · dcb2c00 · dcb2c00
1 parent 6ca02d1
commit dcb2c00
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/src/nnabla/utils/converter/functions.yaml b/python/src/nnabla/utils/converter/functions.yaml
@@ -363,8 +363,9 @@ Neural Network Layer:
       x0:
         doc: Indices with shape :math:`(I_0, ..., I_N)`
         template: TI
-      x1:
+      w:
         doc: Weights with shape :math:`(W_0, ..., W_M)`
+        parameter: true
     outputs:
       y:
         doc: Output with shape :math:`(I_0, ..., I_N, W_1, ..., W_M)`