diff options
| author |  Sami Kama <skama@nvidia.com> | 2018-06-19 11:51:52 -0700 |
|---|---|---|
| committer | Sami Kama <skama@nvidia.com> | 2018-06-19 11:51:52 -0700 |
| commit | 0fb21f608c334dfcaadab7b918c06b88afa8c592 (patch) | |
| tree | 15e0aee8cf1f179de8eaf75a6537201287eabd4c | |
| parent | 941dd4d4ae6d4cfa9b70cd061aa207e04e7730ae (diff) | |
Another linter fix
| -rw-r--r-- | tensorflow/contrib/tensorrt/test/test_tftrt.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/tensorrt/test/test_tftrt.py b/tensorflow/contrib/tensorrt/test/test_tftrt.py index 12e84f7d3c..9a031ddf4e 100644 --- a/tensorflow/contrib/tensorrt/test/test_tftrt.py +++ b/tensorflow/contrib/tensorrt/test/test_tftrt.py @@ -236,7 +236,7 @@ def auto(multi_engine): orig_graph = get_simple_graph_def() # use a frozen graph for inference dummy_input = np.random.random_sample(inp_dims) opt_config = rwpb2.RewriterConfig() - opt_config.meta_optimizer_iterations=opt_config.ONE + opt_config.meta_optimizer_iterations = opt_config.ONE opt_config.optimizers.extend(["constfold", "layout"]) custom_op = opt_config.custom_optimizers.add() custom_op.name = "TensorRTOptimizer" |