Skip to content

Commit

Permalink
fix bug
Browse files Browse the repository at this point in the history
  • Loading branch information
dboyliao committed May 9, 2019
1 parent 4fc9b14 commit 1bf33d5
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions utensor_cgen/backend/code_generator.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,12 +5,12 @@
from tempfile import NamedTemporaryFile

import numpy as np

import tensorflow as tf
from tensorflow.core.framework.graph_pb2 import GraphDef
from tensorflow.tools.graph_transforms import TransformGraph

from utensor_cgen.ir import uTensorGraph
from utensor_cgen.frontend import FrontendSelector
from utensor_cgen.ir import uTensorGraph
from utensor_cgen.transformer.optimizer import RefCntOptimizer
from utensor_cgen.transformer.pipline import TransformerPipeline
from utensor_cgen.utils import NamescopedKWArgsParser
Expand Down Expand Up @@ -118,7 +118,7 @@ def formatter(name, kwargs):
container.add_snippet(cmt_snippet)
composer.add_snippet(container)

if 'inline' in self.trans_methods:
if ('inline', {}) in self.trans_methods:
_logger.info("Generate weight file: %s", weightheader_fname)
with open(weightheader_fname, "w") as wf:
wf.write('// Auto generated by utensor-cli\n\n')
Expand Down

0 comments on commit 1bf33d5

Please sign in to comment.