From ad59129c07385b067997297e3edfa72f03ad27bc Mon Sep 17 00:00:00 2001
From: tronzhang <6517937+tronzhang@user.noreply.gitee.com>
Date: Wed, 6 Jan 2021 18:33:32 +0800
Subject: [PATCH] if input is none, make it a null list

---
 mindspore/_extends/graph_kernel/model/model_builder.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/mindspore/_extends/graph_kernel/model/model_builder.py b/mindspore/_extends/graph_kernel/model/model_builder.py
index 2dd2a93fe5..f66cdb1eda 100644
--- a/mindspore/_extends/graph_kernel/model/model_builder.py
+++ b/mindspore/_extends/graph_kernel/model/model_builder.py
@@ -1,4 +1,4 @@
-# Copyright 2020 Huawei Technologies Co., Ltd
+# Copyright 2020-2021 Huawei Technologies Co., Ltd
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -274,7 +274,7 @@ class CompositeGraph:
 
         builder = GraphBuilder()
         with builder.graph_scope(desc['op']):
-            for in_desc in desc['input_desc']:
+            for in_desc in desc['input_desc'] if desc['input_desc'] is not None else []:
                 name, shape, dtype, data_format = in_desc[0]['tensor_name'], in_desc[
                     0]['shape'], in_desc[0]['data_type'], in_desc[0]['format']
                 self.tensors[name] = builder.tensor(