Searched refs:device_ctxt (Results 1 – 2 of 2) sorted by relevance
244 DeviceContext* device_ctxt = ctx->op_device_context(); in ComputeAsync() local258 device_ctxt->CopyDeviceTensorToCPU( in ComputeAsync()301 DeviceContext* device_ctxt = ctx->op_device_context(); in ComputeAsync() local307 device_ctxt->CopyCPUTensorToDevice( in ComputeAsync()
79 DeviceContext* device_ctxt = context->op_device_context(); in Compute() local83 device, device_ctxt, &src_tensor, copied_tensor, in Compute()