Searched refs:dec_num_deferred_ops_function (Results 1 – 5 of 5) sorted by relevance
366 auto dec_num_deferred_ops_function = ctx->dec_num_deferred_ops_function(); in UnregisterCancellation() local405 dec_num_deferred_ops_function(); in UnregisterCancellation()
703 std::function<void()> dec_num_deferred_ops_function; member1263 TF_MUST_USE_RESULT std::function<void()> dec_num_deferred_ops_function() { in dec_num_deferred_ops_function() function1265 return params_->dec_num_deferred_ops_function in dec_num_deferred_ops_function()1266 ? params_->dec_num_deferred_ops_function in dec_num_deferred_ops_function()
271 params.dec_num_deferred_ops_function = [op_execution_state]() { in Run()
302 params->dec_num_deferred_ops_function = nullptr; in PrepOpKernel()
719 params.dec_num_deferred_ops_function = [this]() { in Process()