Home
last modified time | relevance | path

Searched refs:op_dev_ctx (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_reducer.cc66 DeviceContext* op_dev_ctx = col_ctx_->op_ctx->op_device_context(); in Run() local
69 op_dev_ctx->CopyCPUTensorToDevice( in Run()
/external/tensorflow/tensorflow/core/common_runtime/
Dhierarchical_tree_broadcaster.cc376 DeviceContext* op_dev_ctx = col_ctx_->op_ctx->op_device_context(); in RunTree() local
378 op_dev_ctx, op_dev_ctx, col_ctx_->device, col_ctx_->device, in RunTree()
Dring_reducer.cc143 DeviceContext* op_dev_ctx = col_ctx_->op_ctx->op_device_context(); in ContinueAfterInputCopy() local
144 op_dev_ctx->CopyCPUTensorToDevice( in ContinueAfterInputCopy()