[WebGL] improve RESHAPE

intel · Dec 5, 2018 · af02c9c · af02c9c
1 parent d9788ef
commit af02c9c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/nn/webgl/WebGLModel.js b/src/nn/webgl/WebGLModel.js
@@ -244,7 +244,10 @@ export default class WebGLModel {
         const input = operands[inputs[0]];
         const targetShape = operands[inputs[1]];
         const output = operands[outputs[0]];
-        output.assign(input.reshape(targetShape.dataSync()));
+        if (targetShape.value === undefined) {
+          targetShape.value = targetShape.dataSync();
+        }
+        output.assign(input.reshape(targetShape.value));
       } break;
       case OperationCode.CONCATENATION: {
         const numInputTensors = inputs.length - 1;