Searched refs:device_locality (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | permuter.cc | 60 &col_ctx->device_locality); in InitializeCollectiveContext()
92 col_ctx_->op_ctx->output_alloc_attr(0), tensor, col_ctx_->device_locality, in DispatchSend()
109 col_ctx_->op_ctx->output_alloc_attr(0), tensor, col_ctx_->device_locality, in DispatchRecv()
|
| D | collective_util.cc | 34 DeviceLocality* device_locality) { in InitializeDeviceAndLocality() argument
43 *device_locality = (*device)->attributes().locality(); in InitializeDeviceAndLocality()
|
| D | collective_util.h | 32 DeviceLocality* device_locality);
|
| D | hierarchical_tree_broadcaster.cc | 194 &col_ctx->device_locality); in InitializeCollectiveContext()
426 col_ctx_->device_locality, col_ctx_->op_ctx->cancellation_manager(), in DispatchSend()
447 col_ctx_->device_locality, 0 /*stream_index*/, in DispatchRecv()
|
| D | ring_alg.cc | 251 &col_ctx->device_locality); in InitializeCollectiveContext()
397 col_ctx_->device_locality, col_ctx_->op_ctx->cancellation_manager(), in DispatchSend()
418 col_ctx_->device_locality, rf->subdiv_idx, in DispatchRecv()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl.cc | 67 &col_ctx->device_locality); in InitializeCollectiveContext()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 377 DeviceLocality device_locality; member
|