Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
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()
Dhlo_pass_interface.h101 HloModule* module, RunState* run_state, in RunOnChangedComputations()
/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
Dkernel_fallback_utils.cc29 tfrt_stub::OpKernelRunState& run_state) { in SetUpParams()
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
/external/tensorflow/tensorflow/core/tfrt/fallback/
Dop_kernel_runner_test.cc68 tfrt_stub::OpKernelRunState run_state(new_inputs, params); in TEST() local
/external/tensorflow/tensorflow/lite/delegates/flex/
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
/external/tensorflow/tensorflow/core/common_runtime/
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()
/external/cronet/base/message_loop/
Dmessage_pump_libevent.cc281 RunState run_state(delegate); in Run() local
Dmessage_pump_epoll.cc88 RunState run_state(delegate); in Run() local
Dmessage_pump_win.cc73 RunState run_state(delegate); in Run() local
/external/tensorflow/tensorflow/core/distributed_runtime/
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