Searched defs:from_device (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_rma_local.cc | 41 Device* from_device; in RecvFromPeer() local 99 Device* from_device, DeviceContext* from_device_ctx, in PostToPeer()
|
| D | permuter_test.cc | 90 const string& key, Device* from_device, in PostToPeer()
|
| D | ring_gatherer_test.cc | 83 const string& key, Device* from_device, in PostToPeer()
|
| D | ring_reducer_test.cc | 84 const string& key, Device* from_device, in PostToPeer()
|
| D | hierarchical_tree_broadcaster_test.cc | 180 const string& key, Device* from_device, in PostToPeer()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_module_group_metadata.cc | 541 auto from_device = GetInstructionDevice(*channel.send); in DumpCollectedStats() local
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | core_test.py | 1074 def _recv(self, dtype, tensor_name, from_device): argument
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_driver.cc | 1577 auto from_device = DeviceFromContext(from); in CanEnablePeerAccess() local
|