Browse Source

!454 fix com

Merge pull request !454 from 于佼良/master
pull/455/MERGE
i-robot Gitee 3 years ago
parent
commit
1964175f89
1 changed files with 4 additions and 1 deletions
  1. +4
    -1
      parser/func_to_graph/func2graph.py

+ 4
- 1
parser/func_to_graph/func2graph.py View File

@@ -11,7 +11,7 @@ import os
import sys
import getopt

import graph_library_pb2
from google.protobuf import text_format
import tensorflow as tf
from tensorflow.python.framework.errors_impl import NotFoundError
from tensorflow.python.platform import gfile
@@ -26,6 +26,9 @@ from tensorflow.python.framework import versions

sys.path.append(os.path.join(os.path.split(os.path.realpath(__file__))[0], "util"))

import graph_library_pb2


def _get_num_args(arg_def, node_def):
if arg_def.number_attr:
return node_def.attr[arg_def.number_attr].i


Loading…
Cancel
Save