Searched defs:from_device (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | base_collective_executor.h | 133 const string& key, Device* from_device, in PostToPeer()
|
| D | collective_rma_local.cc | 73 Device* from_device, DeviceContext* from_device_ctx, in PostToPeer()
|
| D | test_collective_executor_mgr.h | 42 const string& key, Device* from_device, in PostToPeer()
|
| D | ring_gatherer_test.cc | 81 const string& key, Device* from_device, in PostToPeer()
|
| D | ring_reducer_test.cc | 81 const string& key, Device* from_device, in PostToPeer()
|
| D | hierarchical_tree_broadcaster_test.cc | 175 const string& key, Device* from_device, in PostToPeer()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_module_group_metadata.cc | 523 auto from_device = GetInstructionDevice(*channel.send); in DumpCollectedStats() local
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | core_test.py | 717 def _recv(self, dtype, tensor_name, from_device): argument
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_driver.cc | 1571 auto from_device = DeviceFromContext(from); in CanEnablePeerAccess() local
|