Searched refs:device_ctx (Results 1 – 1 of 1) sorted by relevance
418 DeviceContext* device_ctx = ctx_->op_device_context(); in StartBody() local421 device_ctx->CopyDeviceTensorToCPU( in StartBody()