Searched refs:device_locality (Results 1 – 6 of 6) sorted by relevance
34 DeviceLocality* device_locality) { in InitializeDeviceAndLocality() argument43 *device_locality = (*device)->attributes().locality(); in InitializeDeviceAndLocality()
194 &col_ctx->device_locality); in InitializeCollectiveContext()415 src_tensor, col_ctx_->device_locality, done); in DispatchSend()435 col_ctx_->device_locality, 0 /*stream_index*/, done); in DispatchRecv()
32 DeviceLocality* device_locality);
249 &col_ctx->device_locality); in InitializeCollectiveContext()395 col_ctx_->device_locality, done); in DispatchSend()415 col_ctx_->device_locality, rf->subdiv_idx, done); in DispatchRecv()
49 &col_ctx->device_locality); in InitializeCollectiveContext()
359 DeviceLocality device_locality; variable