Home
last modified time | relevance | path

Searched defs:to_device_ctx (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_rma_distributed.cc37 DeviceContext* to_device_ctx, in RecvBufCall()
77 const string& key, Device* to_device, DeviceContext* to_device_ctx, in RecvFromPeer()
Dcollective_rma_distributed_test.cc276 DeviceContext* to_device_ctx = nullptr; in TEST_F() local
304 DeviceContext* to_device_ctx = nullptr; in TEST_F() local
336 DeviceContext* to_device_ctx = nullptr; in TEST_F() local
/external/tensorflow/tensorflow/contrib/gdr/
Dgdr_collective_executor_mgr.cc38 DeviceContext* to_device_ctx, in RecvBufCall()
80 DeviceContext* to_device_ctx, in RecvFromPeer()
/external/tensorflow/tensorflow/core/common_runtime/
Dbase_collective_executor.h123 DeviceContext* to_device_ctx, in RecvFromPeer()
Dcollective_rma_local.cc28 const string& key, Device* to_device, DeviceContext* to_device_ctx, in RecvFromPeer()
Dtest_collective_executor_mgr.h33 DeviceContext* to_device_ctx, in RecvFromPeer()
Dring_gatherer_test.cc68 DeviceContext* to_device_ctx, in RecvFromPeer()
Dring_reducer_test.cc68 DeviceContext* to_device_ctx, in RecvFromPeer()
Dhierarchical_tree_broadcaster_test.cc164 DeviceContext* to_device_ctx, in RecvFromPeer()