Home
last modified time | relevance | path

Searched refs:dec_num_deferred_ops_function (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/
H A Dtpu_execute.cc370 auto dec_num_deferred_ops_function = ctx->dec_num_deferred_ops_function(); in UnregisterCancellation() local
409 dec_num_deferred_ops_function(); in UnregisterCancellation()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dop_kernel.h675 std::function<void()> dec_num_deferred_ops_function; member
1224 TF_MUST_USE_RESULT std::function<void()> dec_num_deferred_ops_function() { in dec_num_deferred_ops_function() function
1226 return params_->dec_num_deferred_ops_function in dec_num_deferred_ops_function()
1227 ? params_->dec_num_deferred_ops_function in dec_num_deferred_ops_function()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dkernel_and_device.cc293 params.dec_num_deferred_ops_function = [op_execution_state]() { in Run()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/device/
H A Ddevice_event_mgr_test.cc302 params->dec_num_deferred_ops_function = nullptr; in PrepOpKernel()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dexecutor.cc756 params.dec_num_deferred_ops_function = [this]() { in ProcessInline()