Searched refs:attribution_device (Results 1 – 1 of 1) sorted by relevance
996 Device* attribution_device = kernel->device(); in EagerKernelExecute() local997 if (attribution_device == nullptr) { in EagerKernelExecute()998 attribution_device = ctx->HostCPU(); in EagerKernelExecute()1002 if (ctx->devices()->at(i) == attribution_device) { in EagerKernelExecute()1009 dev_stats->set_device(attribution_device->name()); in EagerKernelExecute()