| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_collective_executor_mgr.cc | 38 DeviceContext* to_device_ctx, in RecvBufCall() argument
80 DeviceContext* to_device_ctx, in RecvFromPeer() argument
87 peer_device, peer_task, peer_is_local, key, to_device, to_device_ctx, in RecvFromPeer()
103 to_device_ctx, to_tensor, done](const Status& s) { in RecvFromPeer()
107 to_device_ctx, to_alloc_attr.on_host(), done); in RecvFromPeer()
119 to_device, to_device_ctx, to_alloc_attr, in RecvFromPeer()
126 step_id_, peer_device, peer_task, key, to_device, to_device_ctx, in RecvFromPeer()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed.cc | 37 DeviceContext* to_device_ctx, in RecvBufCall() argument
77 const string& key, Device* to_device, DeviceContext* to_device_ctx, in RecvFromPeer() argument
83 peer_device, peer_task, peer_is_local, key, to_device, to_device_ctx, in RecvFromPeer()
99 to_device_ctx, to_tensor, dev_to_dev_stream_index, in RecvFromPeer()
136 nullptr /*send_dev_ctx*/, to_device_ctx, cpu_dev, in RecvFromPeer()
163 to_device, to_device_ctx, to_alloc_attr, in RecvFromPeer()
170 step_id_, peer_device, peer_task, key, to_device, to_device_ctx, in RecvFromPeer()
|
| D | collective_rma_distributed_test.cc | 276 DeviceContext* to_device_ctx = nullptr; in TEST_F() local
281 kBufKey, dst_device, to_device_ctx, alloc_attr_, &to_tensor_, in TEST_F()
304 DeviceContext* to_device_ctx = nullptr; in TEST_F() local
309 kBufKey, dst_device, to_device_ctx, alloc_attr_, &to_tensor_, in TEST_F()
336 DeviceContext* to_device_ctx = nullptr; in TEST_F() local
341 kBufKey, dst_device, to_device_ctx, alloc_attr_, &to_tensor_, in TEST_F()
|
| D | collective_rma_distributed.h | 37 DeviceContext* to_device_ctx,
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_rma_local.cc | 28 const string& key, Device* to_device, DeviceContext* to_device_ctx, in RecvFromPeer() argument
41 key, [to_tensor, to_device_ctx, to_device, to_alloc_attr, in RecvFromPeer()
51 to_device_ctx, // dst DeviceContext in RecvFromPeer()
|
| D | base_collective_executor.h | 123 DeviceContext* to_device_ctx, in RecvFromPeer() argument
128 peer_device, peer_task, peer_is_local, key, to_device, to_device_ctx, in RecvFromPeer()
|
| D | collective_rma_local.h | 41 DeviceContext* to_device_ctx,
|
| D | test_collective_executor_mgr.h | 33 DeviceContext* to_device_ctx, in RecvFromPeer() argument
|
| D | ring_gatherer_test.cc | 68 DeviceContext* to_device_ctx, in RecvFromPeer() argument
75 peer_device, peer_task, peer_is_local, key, to_device, to_device_ctx, in RecvFromPeer()
|
| D | ring_reducer_test.cc | 68 DeviceContext* to_device_ctx, in RecvFromPeer() argument
75 peer_device, peer_task, peer_is_local, key, to_device, to_device_ctx, in RecvFromPeer()
|
| D | hierarchical_tree_broadcaster_test.cc | 164 DeviceContext* to_device_ctx, in RecvFromPeer() argument
170 peer_device, peer_task, peer_is_local, key, to_device, to_device_ctx, in RecvFromPeer()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 237 Device* to_device, DeviceContext* to_device_ctx,
|