tf.ConfigProto.RegisterExtension(extension_handle)

tf.ConfigProto.RegisterExtension(extension_handle)