Home
last modified time | relevance | path

Searched defs:source_rank (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dhierarchical_tree_broadcaster.cc226 int source_rank = impl.subdiv_source_rank[subdiv]; in TreeRecvFrom() local
246 int source_rank = impl.subdiv_source_rank[subdiv]; in TreeSendTo() local
302 int source_rank = col_params_->instance.impl_details.subdiv_source_rank[si]; in RunTree() local
H A Dring_alg.cc65 int section, int source_rank) { in RingAlgBufKey()
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dcollective_param_resolver_distributed_test.cc166 int source_rank = 0) { in DefineCollectiveParams()
356 const int source_rank = 0; in TEST_F() local
367 const int source_rank = 3; in TEST_F() local
H A Dcollective_param_resolver_distributed.cc329 int32_t source_rank = resp.source_rank(); in UpdateInstanceCache() local
/aosp_15_r20/external/tensorflow/tensorflow/core/nccl/
H A Dnccl_manager.h162 int source_rank; member
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Dworker.proto590 int32 source_rank = 2; field
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dcollective.h151 int source_rank = -1; // broadcast only 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.h10720 inline ::int32_t CompleteInstanceResponse::source_rank() const { in source_rank() function