tf.TensorInfo.RegisterExtension(extension_handle)

tf.TensorInfo.RegisterExtension(extension_handle)