Skip to content

Instantly share code, notes, and snippets.

@arafatkatze
Last active June 22, 2016 18:49
Show Gist options
  • Save arafatkatze/fc6c6f6a66fad105efbaba2baa0dffca to your computer and use it in GitHub Desktop.
Save arafatkatze/fc6c6f6a66fad105efbaba2baa0dffca to your computer and use it in GitHub Desktop.
import tensorflow as tf
import numpy as np
input1 = tf.placeholder(tf.int64, shape=(2, 2), name = "input1")
input2 = tf.placeholder(tf.int64, shape=(2, 2), name = "input2")
output = tf.add(input1, input2, name = "output")
with tf.Session() as sess:
tf.train.write_graph(sess.graph_def, "model/", "graph.pb", as_text=False)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment