Home
last modified time | relevance | path

Searched refs:dev_ctx (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_reducer_test.cc260 DeviceContext* dev_ctx = nullptr; in DoReduce() local
263 dev_ctx = dev_info->default_context; in DoReduce()
264 dev_ctx->Ref(); in DoReduce()
266 dev_ctx = new DeviceContext; in DoReduce()
268 input_dc.push_back(dev_ctx); in DoReduce()
270 op_params.op_device_context = dev_ctx; in DoReduce()
302 dev_ctx->Unref(); in DoReduce()
/external/tensorflow/tensorflow/core/common_runtime/
Dring_gatherer_test.cc453 DeviceContext* dev_ctx = nullptr; in DoGather() local
456 dev_ctx = dev_info->default_context; in DoGather()
457 dev_ctx->Ref(); in DoGather()
459 dev_ctx = new DeviceContext; in DoGather()
461 input_dc.push_back(dev_ctx); in DoGather()
463 op_params.op_device_context = dev_ctx; in DoGather()
495 dev_ctx->Unref(); in DoGather()
Dring_reducer_test.cc482 DeviceContext* dev_ctx = nullptr; in DoReduce() local
485 dev_ctx = dev_info->default_context; in DoReduce()
486 dev_ctx->Ref(); in DoReduce()
488 dev_ctx = new DeviceContext; in DoReduce()
490 input_dc.push_back(dev_ctx); in DoReduce()
492 op_params.op_device_context = dev_ctx; in DoReduce()
524 dev_ctx->Unref(); in DoReduce()
Dbuf_rendezvous.cc66 DeviceContext* dev_ctx, const Tensor* v, in ProvideBuf() argument
91 h->prod_ctx = dev_ctx; in ProvideBuf()
Dhierarchical_tree_broadcaster_test.cc641 DeviceContext* dev_ctx = nullptr; in DoBroadcast() local
644 dev_ctx = dev_info->default_context; in DoBroadcast()
645 dev_ctx->Ref(); in DoBroadcast()
647 dev_ctx = new DeviceContext; in DoBroadcast()
649 input_dc.push_back(dev_ctx); in DoBroadcast()
651 op_params.op_device_context = dev_ctx; in DoBroadcast()
695 dev_ctx->Unref(); in DoBroadcast()
Dbuf_rendezvous.h77 void ProvideBuf(const string& key, Device* dev, DeviceContext* dev_ctx,