TF_CAPI_EXPORT extern bool TFE_CancellationManagerDeregisterCallback(
    TFE_CancellationManager*, TFE_CancellationToken token);
TF_CAPI_EXPORT extern bool TFE_CancellationManagerTryDeregisterCallback(
    TFE_CancellationManager*, TFE_CancellationToken token);
TF_CAPI_EXPORT extern void TFE_DeleteCancellationManager(
    TFE_CancellationManager*);

    TFE_CancellationManager* cancellation_manager,
    TFE_CancellationToken token) {
  return tensorflow::unwrap(cancellation_manager)->DeregisterCallback(token);
}

bool TFE_CancellationManagerTryDeregisterCallback(
    TFE_CancellationManager* cancellation_manager,
    TFE_CancellationToken token) {
  return tensorflow::unwrap(cancellation_manager)->TryDeregisterCallback(token);
}

                                                      "callback3"));

  EXPECT_TRUE(TFE_CancellationManagerDeregisterCallback(c_mgr, token1));
  EXPECT_TRUE(TFE_CancellationManagerTryDeregisterCallback(c_mgr, token3));

  TFE_CancellationManagerStartCancel(c_mgr);
  EXPECT_TRUE(TFE_CancellationManagerIsCancelled(c_mgr));
  EXPECT_TRUE(callback2_invoked);
  TFE_DeleteCancellationManager(c_mgr);
}

Ilya Tikhonovskiy <******@****.***> 1771913400 -0800