Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dop_kernel.h674 std::function<void()> inc_num_deferred_ops_function; member
1218 TF_MUST_USE_RESULT std::function<void()> inc_num_deferred_ops_function() { in inc_num_deferred_ops_function() function
1220 return params_->inc_num_deferred_ops_function in inc_num_deferred_ops_function()
1221 ? params_->inc_num_deferred_ops_function in inc_num_deferred_ops_function()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/
H A Dtpu_execute.cc369 ctx->inc_num_deferred_ops_function()(); in UnregisterCancellation()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dkernel_and_device.cc290 params.inc_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.cc301 params->inc_num_deferred_ops_function = nullptr; in PrepOpKernel()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dexecutor.cc752 params.inc_num_deferred_ops_function = [this]() { in ProcessInline()