Searched refs:from_device_ctx (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_rma_local.cc | 73 Device* from_device, DeviceContext* from_device_ctx, in PostToPeer() argument
78 buf_rendezvous_.ProvideBuf(key, from_device, from_device_ctx, from_tensor, in PostToPeer()
|
| D | base_collective_executor.h | 134 DeviceContext* from_device_ctx, in PostToPeer() argument
140 from_device_ctx, from_alloc_attr, from_tensor, in PostToPeer()
|
| D | collective_rma_local.h | 49 DeviceContext* from_device_ctx,
|
| D | test_collective_executor_mgr.h | 43 DeviceContext* from_device_ctx, in PostToPeer() argument
|
| D | ring_gatherer_test.cc | 82 DeviceContext* from_device_ctx, in PostToPeer() argument
89 peer_device, peer_task, key, from_device, from_device_ctx, in PostToPeer()
|
| D | ring_reducer_test.cc | 82 DeviceContext* from_device_ctx, in PostToPeer() argument
89 peer_device, peer_task, key, from_device, from_device_ctx, in PostToPeer()
|
| D | hierarchical_tree_broadcaster_test.cc | 176 DeviceContext* from_device_ctx, in PostToPeer() argument
183 peer_device, peer_task, key, from_device, from_device_ctx, in PostToPeer()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 246 DeviceContext* from_device_ctx,
|