Merge pull request !679 from 傅骏/br_build_check_warnningspull/649/MERGE
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import tensorflow as tf | import tensorflow as tf | ||||
| import os | import os | ||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import onnx | import onnx | ||||
| from onnx import helper | from onnx import helper | ||||
| from onnx import AttributeProto, TensorProto, GraphProto | from onnx import AttributeProto, TensorProto, GraphProto | ||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import os | import os | ||||
| import numpy as np | import numpy as np | ||||
| import onnx | import onnx | ||||
| @@ -42,4 +46,4 @@ def gen_onnx(): | |||||
| print(model_def) | print(model_def) | ||||
| if __name__ == "__main__": | if __name__ == "__main__": | ||||
| gen_onnx() | |||||
| gen_onnx() | |||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import tensorflow as tf | import tensorflow as tf | ||||
| import os | import os | ||||
| from tensorflow.python.framework import graph_util | from tensorflow.python.framework import graph_util | ||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import tensorflow as tf | import tensorflow as tf | ||||
| import numpy as np | import numpy as np | ||||
| @@ -11,4 +15,4 @@ def generate_VarIsInitializedOp_pb(): | |||||
| tf.io.write_graph(sess.graph, logdir="./", name="test_VarIsInitializedOp.pb", as_text=False) | tf.io.write_graph(sess.graph, logdir="./", name="test_VarIsInitializedOp.pb", as_text=False) | ||||
| if __name__=='__main__': | if __name__=='__main__': | ||||
| generate_VarIsInitializedOp_pb() | |||||
| generate_VarIsInitializedOp_pb() | |||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import tensorflow as tf | import tensorflow as tf | ||||
| import os | import os | ||||
| import numpy as np | import numpy as np | ||||
| @@ -38,4 +42,4 @@ def generate_case_2(): | |||||
| tf.io.write_graph(sess.graph, logdir="./", name="avgpool3dgrad.pb.txt", as_text=False) | tf.io.write_graph(sess.graph, logdir="./", name="avgpool3dgrad.pb.txt", as_text=False) | ||||
| if __name__=='__main__': | if __name__=='__main__': | ||||
| generate_case_2() | |||||
| generate_case_2() | |||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import tensorflow as tf | import tensorflow as tf | ||||
| import os | import os | ||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import tensorflow as tf | import tensorflow as tf | ||||
| import numpy as np | import numpy as np | ||||
| from tensorflow.python.framework import graph_util | from tensorflow.python.framework import graph_util | ||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import tensorflow as tf | import tensorflow as tf | ||||
| import os | import os | ||||
| from tensorflow.python.framework import graph_util | from tensorflow.python.framework import graph_util | ||||
| @@ -23,4 +27,4 @@ def generate_add_pb(): | |||||
| if __name__=='__main__': | if __name__=='__main__': | ||||
| generate_conv2d_pb() | generate_conv2d_pb() | ||||
| generate_add_pb() | |||||
| generate_add_pb() | |||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import tensorflow as tf | import tensorflow as tf | ||||
| from tensorflow.python.ops import control_flow_ops | from tensorflow.python.ops import control_flow_ops | ||||
| @@ -10,4 +14,4 @@ def generate_enter_pb(): | |||||
| tf.io.write_graph(sess.graph, logdir="./", name="test_enter.pb", as_text=False) | tf.io.write_graph(sess.graph, logdir="./", name="test_enter.pb", as_text=False) | ||||
| if __name__=='__main__': | if __name__=='__main__': | ||||
| generate_enter_pb() | |||||
| generate_enter_pb() | |||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import tensorflow as tf | import tensorflow as tf | ||||
| import numpy as np | import numpy as np | ||||
| @@ -9,4 +13,4 @@ def generate_fill_pb(): | |||||
| tf.io.write_graph(sess.graph, logdir="./", name="test_fill.pb", as_text=False) | tf.io.write_graph(sess.graph, logdir="./", name="test_fill.pb", as_text=False) | ||||
| if __name__ == "__main__": | if __name__ == "__main__": | ||||
| generate_fill_pb() | |||||
| generate_fill_pb() | |||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import tensorflow as tf | import tensorflow as tf | ||||
| def generate_identity_pb(): | def generate_identity_pb(): | ||||
| @@ -10,4 +14,4 @@ def generate_identity_pb(): | |||||
| tf.io.write_graph(sess.graph, logdir="./", name="test_identity.pb", as_text=False) | tf.io.write_graph(sess.graph, logdir="./", name="test_identity.pb", as_text=False) | ||||
| if __name__=='__main__': | if __name__=='__main__': | ||||
| generate_identity_pb() | |||||
| generate_identity_pb() | |||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import tensorflow as tf | import tensorflow as tf | ||||
| from tensorflow.python.framework import graph_util | from tensorflow.python.framework import graph_util | ||||
| import numpy as np | import numpy as np | ||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import tensorflow as tf | import tensorflow as tf | ||||
| import numpy as np | import numpy as np | ||||
| @@ -11,4 +15,4 @@ def generate_no_op_pb(): | |||||
| tf.io.write_graph(sess.graph, logdir="./", name="test_no_op.pb", as_text=False) | tf.io.write_graph(sess.graph, logdir="./", name="test_no_op.pb", as_text=False) | ||||
| if __name__=='__main__': | if __name__=='__main__': | ||||
| generate_no_op_pb() | |||||
| generate_no_op_pb() | |||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import tensorflow as tf | import tensorflow as tf | ||||
| def generate_reshape_pb(): | def generate_reshape_pb(): | ||||
| @@ -7,4 +11,4 @@ def generate_reshape_pb(): | |||||
| tf.io.write_graph(sess.graph, logdir="./", name="test_reshape.pb", as_text=False) | tf.io.write_graph(sess.graph, logdir="./", name="test_reshape.pb", as_text=False) | ||||
| if __name__ == "__main__": | if __name__ == "__main__": | ||||
| generate_reshape_pb() | |||||
| generate_reshape_pb() | |||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import tensorflow as tf | import tensorflow as tf | ||||
| import numpy as np | import numpy as np | ||||
| @@ -10,4 +14,4 @@ def generate_sequeeze_pb(): | |||||
| tf.io.write_graph(sess.graph, logdir="./", name="test_sequeeze.pb", as_text=False) | tf.io.write_graph(sess.graph, logdir="./", name="test_sequeeze.pb", as_text=False) | ||||
| if __name__ == "__main__": | if __name__ == "__main__": | ||||
| generate_sequeeze_pb() | |||||
| generate_sequeeze_pb() | |||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import tensorflow as tf | import tensorflow as tf | ||||
| import numpy as np | import numpy as np | ||||
| @@ -8,4 +12,4 @@ def generate_shape_n_pb(): | |||||
| tf.io.write_graph(sess.graph, logdir="./", name="test_shape_n.pb", as_text=False) | tf.io.write_graph(sess.graph, logdir="./", name="test_shape_n.pb", as_text=False) | ||||
| if __name__ == "__main__": | if __name__ == "__main__": | ||||
| generate_shape_n_pb() | |||||
| generate_shape_n_pb() | |||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import tensorflow as tf | import tensorflow as tf | ||||
| from tensorflow.python.ops import control_flow_ops | from tensorflow.python.ops import control_flow_ops | ||||
| @@ -10,4 +14,4 @@ def generate_switch_pb(): | |||||
| tf.io.write_graph(sess.graph, logdir="./", name="test_switch.pb", as_text=False) | tf.io.write_graph(sess.graph, logdir="./", name="test_switch.pb", as_text=False) | ||||
| if __name__=='__main__': | if __name__=='__main__': | ||||
| generate_switch_pb() | |||||
| generate_switch_pb() | |||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import tensorflow as tf | import tensorflow as tf | ||||
| def generate_VariableV2_pb(): | def generate_VariableV2_pb(): | ||||
| @@ -10,4 +14,4 @@ def generate_VariableV2_pb(): | |||||
| tf.io.write_graph(sess.graph, logdir="./", name="test_VariableV2.pb", as_text=False) | tf.io.write_graph(sess.graph, logdir="./", name="test_VariableV2.pb", as_text=False) | ||||
| if __name__=='__main__': | if __name__=='__main__': | ||||
| generate_VariableV2_pb() | |||||
| generate_VariableV2_pb() | |||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import onnx | import onnx | ||||
| from onnx import helper | from onnx import helper | ||||
| from onnx import AttributeProto, TensorProto, GraphProto | from onnx import AttributeProto, TensorProto, GraphProto | ||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import tensorflow as tf | import tensorflow as tf | ||||
| import os | import os | ||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import onnx | import onnx | ||||
| from onnx import helper | from onnx import helper | ||||
| from onnx import AttributeProto, TensorProto, GraphProto | from onnx import AttributeProto, TensorProto, GraphProto | ||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import tensorflow as tf | import tensorflow as tf | ||||
| import os | import os | ||||
| from tensorflow.python.framework import graph_util | from tensorflow.python.framework import graph_util | ||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import tensorflow as tf | import tensorflow as tf | ||||
| import numpy as np | import numpy as np | ||||
| @@ -11,4 +15,4 @@ def generate_VarIsInitializedOp_pb(): | |||||
| tf.io.write_graph(sess.graph, logdir="./", name="test_VarIsInitializedOp.pb", as_text=False) | tf.io.write_graph(sess.graph, logdir="./", name="test_VarIsInitializedOp.pb", as_text=False) | ||||
| if __name__=='__main__': | if __name__=='__main__': | ||||
| generate_VarIsInitializedOp_pb() | |||||
| generate_VarIsInitializedOp_pb() | |||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import tensorflow as tf | import tensorflow as tf | ||||
| import os | import os | ||||
| import numpy as np | import numpy as np | ||||
| @@ -38,4 +42,4 @@ def generate_case_2(): | |||||
| tf.io.write_graph(sess.graph, logdir="./", name="avgpool3dgrad.pb.txt", as_text=False) | tf.io.write_graph(sess.graph, logdir="./", name="avgpool3dgrad.pb.txt", as_text=False) | ||||
| if __name__=='__main__': | if __name__=='__main__': | ||||
| generate_case_2() | |||||
| generate_case_2() | |||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import tensorflow as tf | import tensorflow as tf | ||||
| import os | import os | ||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import tensorflow as tf | import tensorflow as tf | ||||
| import numpy as np | import numpy as np | ||||
| from tensorflow.python.framework import graph_util | from tensorflow.python.framework import graph_util | ||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import tensorflow as tf | import tensorflow as tf | ||||
| import os | import os | ||||
| from tensorflow.python.framework import graph_util | from tensorflow.python.framework import graph_util | ||||
| @@ -23,4 +27,4 @@ def generate_add_pb(): | |||||
| if __name__=='__main__': | if __name__=='__main__': | ||||
| generate_conv2d_pb() | generate_conv2d_pb() | ||||
| generate_add_pb() | |||||
| generate_add_pb() | |||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import tensorflow as tf | import tensorflow as tf | ||||
| from tensorflow.python.ops import control_flow_ops | from tensorflow.python.ops import control_flow_ops | ||||
| @@ -10,4 +14,4 @@ def generate_enter_pb(): | |||||
| tf.io.write_graph(sess.graph, logdir="./", name="test_enter.pb", as_text=False) | tf.io.write_graph(sess.graph, logdir="./", name="test_enter.pb", as_text=False) | ||||
| if __name__=='__main__': | if __name__=='__main__': | ||||
| generate_enter_pb() | |||||
| generate_enter_pb() | |||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import tensorflow as tf | import tensorflow as tf | ||||
| import numpy as np | import numpy as np | ||||
| @@ -9,4 +13,4 @@ def generate_fill_pb(): | |||||
| tf.io.write_graph(sess.graph, logdir="./", name="test_fill.pb", as_text=False) | tf.io.write_graph(sess.graph, logdir="./", name="test_fill.pb", as_text=False) | ||||
| if __name__ == "__main__": | if __name__ == "__main__": | ||||
| generate_fill_pb() | |||||
| generate_fill_pb() | |||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import tensorflow as tf | import tensorflow as tf | ||||
| def generate_identity_pb(): | def generate_identity_pb(): | ||||
| @@ -10,4 +14,4 @@ def generate_identity_pb(): | |||||
| tf.io.write_graph(sess.graph, logdir="./", name="test_identity.pb", as_text=False) | tf.io.write_graph(sess.graph, logdir="./", name="test_identity.pb", as_text=False) | ||||
| if __name__=='__main__': | if __name__=='__main__': | ||||
| generate_identity_pb() | |||||
| generate_identity_pb() | |||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import tensorflow as tf | import tensorflow as tf | ||||
| from tensorflow.python.framework import graph_util | from tensorflow.python.framework import graph_util | ||||
| import numpy as np | import numpy as np | ||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import tensorflow as tf | import tensorflow as tf | ||||
| import numpy as np | import numpy as np | ||||
| @@ -11,4 +15,4 @@ def generate_no_op_pb(): | |||||
| tf.io.write_graph(sess.graph, logdir="./", name="test_no_op.pb", as_text=False) | tf.io.write_graph(sess.graph, logdir="./", name="test_no_op.pb", as_text=False) | ||||
| if __name__=='__main__': | if __name__=='__main__': | ||||
| generate_no_op_pb() | |||||
| generate_no_op_pb() | |||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import tensorflow as tf | import tensorflow as tf | ||||
| def generate_reshape_pb(): | def generate_reshape_pb(): | ||||
| @@ -7,4 +11,4 @@ def generate_reshape_pb(): | |||||
| tf.io.write_graph(sess.graph, logdir="./", name="test_reshape.pb", as_text=False) | tf.io.write_graph(sess.graph, logdir="./", name="test_reshape.pb", as_text=False) | ||||
| if __name__ == "__main__": | if __name__ == "__main__": | ||||
| generate_reshape_pb() | |||||
| generate_reshape_pb() | |||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import tensorflow as tf | import tensorflow as tf | ||||
| import numpy as np | import numpy as np | ||||
| @@ -10,4 +14,4 @@ def generate_sequeeze_pb(): | |||||
| tf.io.write_graph(sess.graph, logdir="./", name="test_sequeeze.pb", as_text=False) | tf.io.write_graph(sess.graph, logdir="./", name="test_sequeeze.pb", as_text=False) | ||||
| if __name__ == "__main__": | if __name__ == "__main__": | ||||
| generate_sequeeze_pb() | |||||
| generate_sequeeze_pb() | |||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import tensorflow as tf | import tensorflow as tf | ||||
| import numpy as np | import numpy as np | ||||
| @@ -8,4 +12,4 @@ def generate_shape_n_pb(): | |||||
| tf.io.write_graph(sess.graph, logdir="./", name="test_shape_n.pb", as_text=False) | tf.io.write_graph(sess.graph, logdir="./", name="test_shape_n.pb", as_text=False) | ||||
| if __name__ == "__main__": | if __name__ == "__main__": | ||||
| generate_shape_n_pb() | |||||
| generate_shape_n_pb() | |||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import tensorflow as tf | import tensorflow as tf | ||||
| from tensorflow.python.ops import control_flow_ops | from tensorflow.python.ops import control_flow_ops | ||||
| @@ -10,4 +14,4 @@ def generate_switch_pb(): | |||||
| tf.io.write_graph(sess.graph, logdir="./", name="test_switch.pb", as_text=False) | tf.io.write_graph(sess.graph, logdir="./", name="test_switch.pb", as_text=False) | ||||
| if __name__=='__main__': | if __name__=='__main__': | ||||
| generate_switch_pb() | |||||
| generate_switch_pb() | |||||
| @@ -1,3 +1,7 @@ | |||||
| #!/usr/bin/env python3 | |||||
| # -*- coding utf-8 -*- | |||||
| # Copyright Huawei Technologies Co., Ltd 2019-2022. All rights reserved. | |||||
| import tensorflow as tf | import tensorflow as tf | ||||
| def generate_VariableV2_pb(): | def generate_VariableV2_pb(): | ||||
| @@ -10,4 +14,4 @@ def generate_VariableV2_pb(): | |||||
| tf.io.write_graph(sess.graph, logdir="./", name="test_VariableV2.pb", as_text=False) | tf.io.write_graph(sess.graph, logdir="./", name="test_VariableV2.pb", as_text=False) | ||||
| if __name__=='__main__': | if __name__=='__main__': | ||||
| generate_VariableV2_pb() | |||||
| generate_VariableV2_pb() | |||||