Home
last modified time | relevance | path

Searched defs:DispatchRecv (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dpermuter.cc96 void Permuter::DispatchRecv(int src_rank, int target_rank, Tensor* tensor, in DispatchRecv() function in tensorflow::Permuter
H A Dall_to_all.cc135 void AllToAll::DispatchRecv(int src_rank, int target_rank, Tensor* tensor, in DispatchRecv() function in tensorflow::AllToAll
H A Dring_alg.cc416 void RingAlg::DispatchRecv(RingField* rf, const StatusCallback& done) { in DispatchRecv() function in tensorflow::RingAlg
H A Dhierarchical_tree_broadcaster.cc433 void HierarchicalTreeBroadcaster::DispatchRecv(int subdiv, int src_rank, in DispatchRecv() function in tensorflow::HierarchicalTreeBroadcaster