Home
last modified time | relevance | path

Searched defs:communicator_key (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/nccl/
H A Dnccl_manager_test.cc319 const string communicator_key = in RunMultiNodeAllReduceTest() local
370 const string communicator_key = in RunMultiNodeBroadcastTest() local
636 const string communicator_key = in TEST() local
877 const string& communicator_key) { in TYPED_TEST()
896 string communicator_key = nodes[0].nccl_manager.GenerateCommunicatorKey(); in TYPED_TEST() local
H A Dnccl_manager.h159 const string& communicator_key; member
H A Dnccl_manager.cc195 const string communicator_key; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dxla_helpers.cc247 const std::string& communicator_key = in ResolveDeviceAssignment() local
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Dworker.proto562 bytes communicator_key = 7; field
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dcollective.h58 string communicator_key; // for communicator-based techniques e.g. NCCL member
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/protobuf/
Dworker.pb.h10225 inline const std::string& CompleteGroupResponse::communicator_key() const { in communicator_key() function
10256 inline void CompleteGroupResponse::set_allocated_communicator_key(std::string* communicator_key) { in set_allocated_communicator_key()