Home
last modified time | relevance | path

Searched refs:MemCpyAsync (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_rma_local.cc50 MemCpyAsync(hook->prod_ctx, // src DeviceContext in RecvFromPeer()
83 void CollectiveRemoteAccessLocal::MemCpyAsync( in MemCpyAsync() function in tensorflow::CollectiveRemoteAccessLocal
Dcollective_rma_local.h76 static void MemCpyAsync(DeviceContext* src_dev_ctx,
Dring_gatherer.cc102 CollectiveRemoteAccessLocal::MemCpyAsync( in Run()
Dring_reducer.cc90 CollectiveRemoteAccessLocal::MemCpyAsync( in Run()
Dhierarchical_tree_broadcaster.cc370 CollectiveRemoteAccessLocal::MemCpyAsync( in RunTree()