Home
last modified time | relevance | path

Searched refs:coordination_service_agent (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/coordination/
DBUILD98 name = "coordination_service_agent",
99 srcs = ["coordination_service_agent.cc"],
100 hdrs = ["coordination_service_agent.h"],
121 ":coordination_service_agent",
141 ":coordination_service_agent",
166 ":coordination_service_agent",
182 ":coordination_service_agent",
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.cc269 CoordinationServiceAgent* coordination_service_agent) { in Run() argument
309 params.coordination_service_agent = coordination_service_agent; in Run()
357 CoordinationServiceAgent* coordination_service_agent) { in PrepareForRun() argument
406 opts->coordination_service_agent = coordination_service_agent; in PrepareForRun()
418 CoordinationServiceAgent* coordination_service_agent) { in Run() argument
426 eager_func_params, coordination_service_agent, in Run()
436 eager_func_params, stack_trace, coordination_service_agent); in Run()
462 CoordinationServiceAgent* coordination_service_agent, in RunAsync() argument
468 absl::nullopt, coordination_service_agent); in RunAsync()
Dkernel_and_device.h143 CoordinationServiceAgent* coordination_service_agent) = 0;
158 CoordinationServiceAgent* coordination_service_agent,
221 CoordinationServiceAgent* coordination_service_agent) override;
228 CoordinationServiceAgent* coordination_service_agent, in RunAsync() argument
232 eager_func_params, {}, coordination_service_agent)); in RunAsync()
323 CoordinationServiceAgent* coordination_service_agent) override;
330 CoordinationServiceAgent* coordination_service_agent,
353 CoordinationServiceAgent* coordination_service_agent);
DBUILD194 "//tensorflow/core/distributed_runtime/coordination:coordination_service_agent",
/external/tensorflow/tensorflow/core/distributed_runtime/integration_test/
DBUILD21 "//tensorflow/core/distributed_runtime/coordination:coordination_service_agent",
49 "//tensorflow/core/distributed_runtime/coordination:coordination_service_agent",
75 "//tensorflow/core/distributed_runtime/coordination:coordination_service_agent",
105 "//tensorflow/core/distributed_runtime/coordination:coordination_service_agent",
137 "//tensorflow/core/distributed_runtime/coordination:coordination_service_agent",
Dcoordination_test_opkernel_registration.cc50 auto* coord_agent = ctx->coordination_service_agent(); in Compute()
90 auto* coord_agent = ctx->coordination_service_agent(); in Compute()
/external/tensorflow/tensorflow/core/distributed_runtime/preemption/
DBUILD19 "//tensorflow/core/distributed_runtime/coordination:coordination_service_agent",
63 "//tensorflow/core/distributed_runtime/coordination:coordination_service_agent",
108 "//tensorflow/core/distributed_runtime/coordination:coordination_service_agent",
Dcheck_preemption_op_kernel.cc37 ctx->coordination_service_agent()->TryGetKeyValue(preemption_key_); in Compute()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dgraph_mgr.h97 CoordinationServiceAgent* coordination_service_agent,
174 CoordinationServiceAgent* coordination_service_agent,
Dgraph_mgr.cc424 CoordinationServiceAgent* coordination_service_agent, StatusCallback done) { in ExecuteAsync() argument
496 coordination_service_agent, in ExecuteAsync()
521 CoordinationServiceAgent* coordination_service_agent, StatusCallback done) { in StartParallelExecutors() argument
545 args.coordination_service_agent = coordination_service_agent; in StartParallelExecutors()
DBUILD159 "//tensorflow/core/distributed_runtime/coordination:coordination_service_agent",
/external/tensorflow/tensorflow/core/common_runtime/
Dexecutor.h105 CoordinationServiceAgent* coordination_service_agent = nullptr; member
Dexecutor.cc437 coordination_service_agent_(args.coordination_service_agent), in ExecutorState()
742 params.coordination_service_agent = coordination_service_agent_; in ProcessInline()
Dfunction.cc1028 exec_args->coordination_service_agent = run_opts.coordination_service_agent; in ExecutorArgsFromOptions()
/external/tensorflow/tensorflow/compiler/xla/pjrt/distributed/
DBUILD101 "//tensorflow/core/distributed_runtime/coordination:coordination_service_agent",
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.h684 CoordinationServiceAgent* coordination_service_agent = nullptr; member
1156 CoordinationServiceAgent* coordination_service_agent() const { in coordination_service_agent() function
1157 return params_->coordination_service_agent; in coordination_service_agent()
Dfunction.h850 CoordinationServiceAgent* coordination_service_agent = nullptr; member
/external/tensorflow/tensorflow/c/eager/
DBUILD898 "//tensorflow/core/distributed_runtime/coordination:coordination_service_agent",