Home
last modified time | relevance | path

Searched defs:run_state (Results 1 – 13 of 13) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_pass_fix.h44 RunState run_state; in RunOnChangedComputations() local
88 HloModule* module, RunState* run_state, in RunToFixPoint()
111 HloModule* module, RunState* run_state, in RunOnChangedComputationsOnce()
H A Dhlo_pass_interface.h101 HloModule* module, RunState* run_state, in RunOnChangedComputations()
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
H A Dkernel_fallback_execute_compat.cc215 thread_local OpKernelRunState run_state; in GetThreadLocalOpKernelRunState() local
227 const tfrt::ExecutionContext& exec_ctx, OpKernelRunState* run_state, in KernelFallbackExecuteCompatAsyncInternal()
249 OpKernelRunState run_state; in KernelFallbackExecuteCompatAsyncInternal() member
301 OpKernelRunState* run_state, const OpKernelRunner& kernel_runner, in KernelFallbackExecuteCompatSyncInternal()
349 auto& run_state = GetThreadLocalOpKernelRunState(); in KernelFallbackExecuteCompatCoreRuntimeDispatch() local
512 auto& run_state = GetThreadLocalOpKernelRunState(); in KernelFallbackExecuteOpInternal() local
H A Dkernel_fallback_utils.cc29 tfrt_stub::OpKernelRunState& run_state) { in SetUpParams()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Ddirect_session.cc514 RunState run_state(step_id, &devices_); in RunInternal() local
980 PartialRunState* run_state = in PRunSetup() local
1044 PartialRunState* run_state; in PRun() local
1204 const ExecutorsAndKeys* executors_and_keys, PartialRunState* run_state, in RecvPRunOutputs()
1250 const PartialRunState* run_state) { in CheckFetch()
1826 void DirectSession::WaitForNotification(Notification* n, RunState* run_state, in WaitForNotification()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/
H A Dkernel.cc261 tensorflow::tfrt_stub::OpKernelRunState* run_state) { in BuildOpKernelInputs()
406 tensorflow::tfrt_stub::OpKernelRunState* run_state, TfLiteContext* context, in ExecuteOpKernelRunner()
731 tensorflow::tfrt_stub::OpKernelRunState run_state; in Eval() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/fallback/
H A Dop_kernel_runner_test.cc68 tfrt_stub::OpKernelRunState run_state(new_inputs, params); in TEST() local
/aosp_15_r20/external/cronet/base/message_loop/
H A Dmessage_pump_fuchsia.cc284 RunState run_state(delegate); in Run() local
H A Dmessage_pump_libevent.cc274 RunState run_state(delegate); in Run() local
H A Dmessage_pump_epoll.cc89 RunState run_state(delegate); in Run() local
H A Dmessage_pump_win.cc77 RunState run_state(delegate); in Run() local
/aosp_15_r20/external/coreboot/src/lib/
H A Dhardwaremain.c53 boot_state_t (*run_state)(void *arg); member
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dmaster_session.cc1038 const RunStepRequestWrapper& req, const RunState* run_state, in CheckFetches()
1642 RunState* run_state = in PartialRunSetup() local
1735 RunState* run_state = nullptr; in DoPartialRun() local