tf.AttrValue.RegisterExtension(extension_handle)

tf.AttrValue.RegisterExtension(extension_handle)