Searched refs:op_dev_ctx (Results 1 – 3 of 3) sorted by relevance
110 DeviceContext* op_dev_ctx = col_ctx_->op_ctx->op_device_context(); in Run() local112 op_dev_ctx->CopyCPUTensorToDevice( in Run()
369 DeviceContext* op_dev_ctx = col_ctx_->op_ctx->op_device_context(); in RunTree() local371 op_dev_ctx, op_dev_ctx, col_ctx_->device, col_ctx_->device, in RunTree()
125 DeviceContext* op_dev_ctx = col_ctx_->op_ctx->op_device_context(); in ContinueAfterInputCopy() local126 op_dev_ctx->CopyCPUTensorToDevice(&group_size_val, col_ctx_->device, in ContinueAfterInputCopy()