tf.train.JobDef.RegisterExtension(extension_handle)

tf.train.JobDef.RegisterExtension(extension_handle)