From 4be4b4c8b5f5f78e54aecbfa336aa5762d1fde13 Mon Sep 17 00:00:00 2001
From: Yinghai Lu <yinghai@fb.com>
Date: Thu, 16 Aug 2018 10:53:27 -0700
Subject: [PATCH] Remove weight from input of onnxifi backend op (#10575)

Summary:
The ONNXIFI backend will absorb the constant weight in Conv, so we should not add it as an input. This is just a test artifacts. Note that Onnxifi transformer will do the right thing when cutting the graph to absorb the weights.

rdzhabarov
Pull Request resolved: https://github.com/pytorch/pytorch/pull/10575

Reviewed By: houseroad

Differential Revision: D9357339

Pulled By: yinghai

fbshipit-source-id: a613fa3acafa687295312f5211f8e9d7f77b39cd
---
 caffe2/python/onnx/test_onnxifi.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/caffe2/python/onnx/test_onnxifi.py b/caffe2/python/onnx/test_onnxifi.py
index 596dcde044d..c1476430ba0 100644
--- a/caffe2/python/onnx/test_onnxifi.py
+++ b/caffe2/python/onnx/test_onnxifi.py
@@ -88,7 +88,7 @@ class OnnxifiTest(TestCase):
         model_def = make_model(graph_def, producer_name='conv-test')
         op = core.CreateOperator(
             "Onnxifi",
-            ["X", "W"],
+            ["X"],
             ["Y"],
             onnx_model=model_def.SerializeToString(),
             initializers=["W", "W"],