fix data type undetermined problem

sony · May 30, 2018 · 22ba80a · 22ba80a
1 parent 2fdcad0
commit 22ba80a
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/build-tools/code_generator/functions.yaml b/build-tools/code_generator/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)`

diff --git a/python/src/nnabla/utils/cli/forward.py b/python/src/nnabla/utils/cli/forward.py
@@ -258,8 +258,13 @@ class ForwardConfig:
     inputs = []
     for e in config.executors:
         for v, d in e.dataset_assign.items():
-            data = np.fromfile(args.inputs[input_file_index], np.float32).reshape(
-                v.variable_instance.d.shape)
+            input_filename = args.inputs[input_file_index]
+            if "int32" in input_filename:
+                data = np.fromfile(input_filename, np.int32).reshape(
+                    v.variable_instance.d.shape)
+            else:
+                data = np.fromfile(input_filename, np.float).reshape(
+                    v.variable_instance.d.shape)
             inputs.append((d, data))
             input_file_index += 1
     data = []