Searched refs:ExitCountdown (Results 1 – 4 of 4) sorted by relevance
105 void ExitCountdown(Env* env, std::shared_ptr<std::atomic<bool>> done) { in ExitCountdown() function137 env->SchedClosure([env, done]() { ExitCountdown(env, done); }); in Compute()
91 void ExitCountdown(Env* env) { in ExitCountdown() function308 [env]() { ExitCountdown(env); }); in TPUCancelExecution()
143 static void ExitCountdown(Env* env, std::shared_ptr<std::atomic<bool>> done);
389 /* static */ void TpuCompileOpKernelCommon::ExitCountdown( in ExitCountdown() function in tensorflow::tpu::TpuCompileOpKernelCommon555 env->SchedClosure([env, done]() { ExitCountdown(env, done); }); in Compute()