tf.RunMetadata.RegisterExtension(extension_handle)

tf.RunMetadata.RegisterExtension(extension_handle)