Home
last modified time | relevance | path

Searched refs:CollectiveExecutorMgrInterface (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/
Dworker_env.h28 class CollectiveExecutorMgrInterface; variable
63 std::unique_ptr<CollectiveExecutorMgrInterface> collective_executor_mgr;
Dmaster_env.h30 class CollectiveExecutorMgrInterface; variable
100 CollectiveExecutorMgrInterface* collective_executor_mgr = nullptr;
/external/tensorflow/tensorflow/core/framework/
Dcollective.h252 class CollectiveExecutorMgrInterface : public StepSequenceInterface {
254 virtual ~CollectiveExecutorMgrInterface() {} in ~CollectiveExecutorMgrInterface()
381 explicit CollectiveExecutor(CollectiveExecutorMgrInterface* cem) in CollectiveExecutor()
386 CollectiveExecutorMgrInterface* cem_;
/external/tensorflow/tensorflow/core/common_runtime/
Dtest_collective_executor_mgr.h30 explicit TestCollectiveExecutor(CollectiveExecutorMgrInterface* cem,
70 class TestCollectiveExecutorMgr : public CollectiveExecutorMgrInterface {
Dcollective_executor_mgr.h26 class CollectiveExecutorMgr : public CollectiveExecutorMgrInterface {
Dbase_collective_executor.h98 BaseCollectiveExecutor(CollectiveExecutorMgrInterface* cem, in BaseCollectiveExecutor()
Ddirect_session.h390 std::unique_ptr<CollectiveExecutorMgrInterface> collective_executor_mgr_;
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcontext.h105 CollectiveExecutorMgrInterface* collective_executor_mgr = nullptr,
340 CollectiveExecutorMgrInterface* collective_executor_mgr() { in collective_executor_mgr()
461 CollectiveExecutorMgrInterface* rpc_collective_executor_mgr);
761 OwnedOrUnownedHelper<CollectiveExecutorMgrInterface> collective_executor_mgr_;
Dcontext.cc152 CollectiveExecutorMgrInterface* collective_executor_mgr, in EagerContext()
1324 CollectiveExecutorMgrInterface* rpc_collective_executor_mgr) { in StoreCollectiveOpsServer()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_server_lib.h51 typedef std::function<CollectiveExecutorMgrInterface*(