Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/distributed_runtime/coordination/
Dcoordination_service_test.cc800 const std::string barrier_id = "barrier_id"; in TEST_F() local
843 const std::string barrier_id = "barrier_id"; in TEST_F() local
873 const std::string barrier_id = "barrier_id"; in TEST_F() local
894 const std::string barrier_id = "barrier_id"; in TEST_F() local
917 const std::string barrier_id = "barrier_id"; in TEST_F() local
939 const std::string barrier_id = "barrier_id"; in TEST_F() local
957 const std::string barrier_id = "barrier_id"; in TEST_F() local
984 const std::string barrier_id = "barrier_id"; in TEST_F() local
1000 const std::string barrier_id = "cancelled_barrier_id"; in TEST_F() local
1016 const std::string barrier_id = "barrier_id"; in TEST_F() local
[all …]
Dcoordination_service.cc330 absl::string_view barrier_id) { in JoinBarrier()
335 absl::string_view barrier_id) { in ExitBarrier()
443 for (const std::string& barrier_id : ongoing_barriers_) { in StartCheckStaleness() local
604 for (const auto& barrier_id : in DisconnectTask() local
907 for (const auto& barrier_id : in SetTaskError() local
920 const std::string& barrier_id, absl::Duration timeout, in BarrierAsync()
1044 const std::string& barrier_id, const CoordinatedTask& task) { in CancelBarrier()
1070 absl::string_view barrier_id, Status result, BarrierState* barrier) { in PassBarrier()
Dcoordination_service_agent.cc673 const std::string& barrier_id, absl::Duration timeout, in WaitAtBarrier()
686 const std::string& barrier_id, absl::Duration timeout, in WaitAtBarrierAsync()
717 const std::string& barrier_id) { in CancelBarrier()
729 const std::string& barrier_id, StatusCallback done) { in CancelBarrierAsync()
Dcoordination_service_agent_test.cc417 const std::string barrier_id = "only_use_once"; in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/pjrt/distributed/
Dclient.cc351 std::string barrier_id, absl::Duration timeout) { in WaitAtBarrier()
545 std::string barrier_id, absl::Duration timeout) { in WaitAtBarrier()
Dprotocol.proto103 bytes barrier_id = 2; field
Dservice.cc428 std::string barrier_id = request->barrier_id(); in WaitAtBarrier() local
Dclient_server_test.cc600 std::string barrier_id; in TEST_P() local
/external/tensorflow/tensorflow/core/protobuf/
Dcoordination_service.proto196 string barrier_id = 1; field
209 string barrier_id = 1; field
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla.cc464 int64_t timeout_in_ms) { in PYBIND11_MODULE()
/external/pytorch/torch/testing/_internal/distributed/
Ddistributed_test.py528 barrier_id = 0 variable in Barrier