| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | rpc_collective_executor_mgr_test.cc | 65 CollectiveExecutor::Handle* h = in TEST_F()
66 new CollectiveExecutor::Handle(cme_->FindOrCreate(1), true); in TEST_F()
68 CollectiveExecutor::Handle* h2 = in TEST_F()
69 new CollectiveExecutor::Handle(cme_->FindOrCreate(1), true); in TEST_F()
71 CollectiveExecutor* ce = h->get(); in TEST_F()
74 CollectiveExecutor* ce2 = cme_->FindOrCreate(1); in TEST_F()
82 EXPECT_EQ(x, CollectiveExecutor::kInvalidId); in TEST_F()
95 EXPECT_NE(x, CollectiveExecutor::kInvalidId); in TEST_F()
127 EXPECT_EQ(x, CollectiveExecutor::kInvalidId); in TEST_F()
129 EXPECT_EQ(y, CollectiveExecutor::kInvalidId); in TEST_F()
[all …]
|
| D | rpc_collective_executor_mgr.cc | 48 CollectiveExecutor* RpcCollectiveExecutorMgr::Create(int64 step_id) { in Create()
152 return CollectiveExecutor::kInvalidId; in NextStepId()
162 it->second->next_step_id_ = CollectiveExecutor::kInvalidId; in RetireStepId()
|
| D | rpc_collective_executor_mgr.h | 59 virtual CollectiveExecutor* Create(int64 step_id) override;
73 : graph_key_(k), next_step_id_(CollectiveExecutor::kInvalidId) {} in GraphKeySequence()
|
| D | graph_mgr.cc | 428 CollectiveExecutor::Handle* ce_handle = in ExecuteAsync()
430 ? new CollectiveExecutor::Handle( in ExecuteAsync()
470 CollectiveExecutor::Handle* ce_handle, in StartParallelExecutors()
|
| D | graph_mgr.h | 167 CollectiveExecutor::Handle* ce_handle,
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_executor_mgr_test.cc | 58 CollectiveExecutor::Handle* h = in TEST_F()
59 new CollectiveExecutor::Handle(cme_->FindOrCreate(1), true); in TEST_F()
61 CollectiveExecutor::Handle* h2 = in TEST_F()
62 new CollectiveExecutor::Handle(cme_->FindOrCreate(1), true); in TEST_F()
64 CollectiveExecutor* ce = h->get(); in TEST_F()
67 CollectiveExecutor::Handle h3(cme_->FindOrCreate(1), true); in TEST_F()
73 EXPECT_EQ(CollectiveExecutor::kInvalidId, cme_->NextStepId(123)); in TEST_F()
|
| D | test_collective_executor_mgr.h | 27 class TestCollectiveExecutor : public CollectiveExecutor {
30 : CollectiveExecutor(cem) {} in TestCollectiveExecutor()
62 CollectiveExecutor* FindOrCreate(int64 step_id) override { in FindOrCreate()
64 CollectiveExecutor* ce = nullptr; in FindOrCreate()
107 return CollectiveExecutor::kInvalidId; in NextStepId()
113 gtl::FlatMap<int64, CollectiveExecutor*> table_ GUARDED_BY(mu_);
|
| D | collective_executor_mgr.h | 33 CollectiveExecutor* FindOrCreate(int64 step_id) override;
53 return CollectiveExecutor::kInvalidId; in NextStepId()
60 virtual CollectiveExecutor* Create(int64 step_id);
70 gtl::FlatMap<int64, CollectiveExecutor*> executor_table_ GUARDED_BY(exec_mu_);
|
| D | collective_executor_mgr.cc | 42 CollectiveExecutor* CollectiveExecutorMgr::FindOrCreate(int64 step_id) { in FindOrCreate()
43 CollectiveExecutor* ce = nullptr; in FindOrCreate()
58 CollectiveExecutor* CollectiveExecutorMgr::Create(int64 step_id) { in Create()
66 CollectiveExecutor* ce = nullptr; in Cleanup()
|
| D | base_collective_executor.h | 94 class BaseCollectiveExecutor : public CollectiveExecutor {
100 : CollectiveExecutor(cem), in BaseCollectiveExecutor()
|
| D | executor.h | 95 CollectiveExecutor* collective_executor = nullptr;
|
| D | direct_session.h | 184 std::unique_ptr<CollectiveExecutor::Handle> collective_executor;
|
| D | ring_gatherer_test.cc | 517 CollectiveExecutor* col_exec_;
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 127 class CollectiveExecutor; variable
216 virtual CollectiveExecutor* FindOrCreate(int64 step_id) = 0;
257 class CollectiveExecutor : public PeerAccessInterface, public core::RefCounted {
300 explicit Handle(CollectiveExecutor* ce, bool inherit_ref) : ce_(ce) { in Handle()
304 CollectiveExecutor* get() const { return ce_; } in get()
307 CollectiveExecutor* ce_;
311 explicit CollectiveExecutor(CollectiveExecutorMgrInterface* cem) in CollectiveExecutor() function
318 TF_DISALLOW_COPY_AND_ASSIGN(CollectiveExecutor);
341 CollectiveContext(CollectiveExecutor* col_exec, const DeviceMgr* dev_mgr,
348 CollectiveExecutor* col_exec; // Not owned
|
| D | collective.cc | 151 /*static*/ OpKernelContext::Params* CollectiveExecutor::CtxParams( in CtxParams()
156 CollectiveContext::CollectiveContext(CollectiveExecutor* col_exec, in CollectiveContext()
176 int64 CollectiveExecutor::kInvalidId = -1;
|
| D | function.h | 39 class CollectiveExecutor; variable
619 CollectiveExecutor* collective_executor = nullptr;
|
| D | op_kernel.h | 75 class CollectiveExecutor; variable
635 CollectiveExecutor* collective_executor = nullptr;
1068 CollectiveExecutor* collective_executor() const { in collective_executor()
1259 friend class CollectiveExecutor; // for access to params_ variable
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.h | 68 std::unique_ptr<CollectiveExecutor::Handle> collective_executor, in KernelAndDevice()
126 const std::unique_ptr<CollectiveExecutor::Handle> collective_executor_;
136 std::unique_ptr<CollectiveExecutor::Handle> collective_executor, in KernelAndDeviceOp()
197 std::unique_ptr<CollectiveExecutor::Handle> collective_executor, in KernelAndDeviceFunc()
|
| D | context.h | 166 std::unique_ptr<CollectiveExecutor::Handle> GetCollectiveExecutorHandle() { in GetCollectiveExecutorHandle()
167 return std::unique_ptr<CollectiveExecutor::Handle>( in GetCollectiveExecutorHandle()
168 new CollectiveExecutor::Handle( in GetCollectiveExecutorHandle()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_ops.cc | 38 bool CanProceedWithCompute(OpKernelContext* c, CollectiveExecutor* col_exec, in CanProceedWithCompute()
88 CollectiveExecutor* col_exec = c->collective_executor(); in ComputeAsync()
186 CollectiveExecutor* col_exec = c->collective_executor(); in ComputeAsync()
251 CollectiveExecutor* col_exec = c->collective_executor(); in ComputeAsync()
320 CollectiveExecutor* col_exec = c->collective_executor(); in ComputeAsync()
|
| D | collective_nccl_reducer_test.cc | 316 CollectiveExecutor* col_exec_;
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_collective_executor_mgr.h | 49 virtual CollectiveExecutor* Create(int64 step_id) override;
|
| D | gdr_collective_executor_mgr.cc | 150 CollectiveExecutor* GdrCollectiveExecutorMgr::Create(int64 step_id) { in Create()
|
| D | gdr_worker.cc | 159 CollectiveExecutor::Handle ce_handle( in RecvBufAsync()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_service.cc | 564 CollectiveExecutor::Handle ce_handle( in RecvBufAsync()
|