Skip to content

Instantly share code, notes, and snippets.

@hadifar

hadifar/tf_checkpoint_15.py

Created December 2, 2018 16:42
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save hadifar/5be3d6bec0a9bdb466ac639877948532 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save hadifar/5be3d6bec0a9bdb466ac639877948532 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
tf_checkpoint_15.py
DEF_SERVING_DEF_KEY = tf.saved_model.signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY
tensor_x = model.signature_def[DEF_SERVING_DEF_KEY].inputs['x'].name # return Placeholder:0
tensor_y = model.signature_def[DEF_SERVING_DEF_KEY].outputs['y'].name # return 'dense/BiasAdd:0'
x = tf.get_default_graph().get_tensor_by_name(tensor_x)
y = tf.get_default_graph().get_tensor_by_name(tensor_y)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment