tf.ConfigProto.DeviceCountEntry.RegisterExtension(extension_handle)

tf.ConfigProto.DeviceCountEntry.RegisterExtension(extension_handle)