Searched defs:to_device_ctx (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed.cc | 37 DeviceContext* to_device_ctx, in RecvBufCall()
77 const string& key, Device* to_device, DeviceContext* to_device_ctx, in RecvFromPeer()
|
| D | collective_rma_distributed_test.cc | 276 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/ |
| D | gdr_collective_executor_mgr.cc | 38 DeviceContext* to_device_ctx, in RecvBufCall()
80 DeviceContext* to_device_ctx, in RecvFromPeer()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | base_collective_executor.h | 123 DeviceContext* to_device_ctx, in RecvFromPeer()
|
| D | collective_rma_local.cc | 28 const string& key, Device* to_device, DeviceContext* to_device_ctx, in RecvFromPeer()
|
| D | test_collective_executor_mgr.h | 33 DeviceContext* to_device_ctx, in RecvFromPeer()
|
| D | ring_gatherer_test.cc | 68 DeviceContext* to_device_ctx, in RecvFromPeer()
|
| D | ring_reducer_test.cc | 68 DeviceContext* to_device_ctx, in RecvFromPeer()
|
| D | hierarchical_tree_broadcaster_test.cc | 164 DeviceContext* to_device_ctx, in RecvFromPeer()
|