Searched refs:device_ctxt (Results 1 – 2 of 2) sorted by relevance
245 DeviceContext* device_ctxt = ctx->op_device_context(); in ComputeAsync() local259 device_ctxt->CopyDeviceTensorToCPU( in ComputeAsync()302 DeviceContext* device_ctxt = ctx->op_device_context(); in ComputeAsync() local308 device_ctxt->CopyCPUTensorToDevice( in ComputeAsync()
89 DeviceContext* device_ctxt = context->op_device_context(); in Compute() local93 device, device_ctxt, &src_tensor, copied_tensor, in Compute()