Searched refs:set_device_name (Results 1 – 3 of 3) sorted by relevance
290 trace->set_device_name(device_name); in WriteGraphExecutionTrace()343 debugged_device->set_device_name(device_name); in RegisterDeviceAndGetId()
314 op->set_device_name(packed_handle->DeviceOrHostCPU(*ctx)->name()); in SerializePackedHandle()423 send_tensor->set_device_name(recv_device_->name()); in StartRemoteSendTensor()
403 void set_device_name(py::handle v) { in set_device_name() function in __anonaa2eb7e00311::EagerContextThreadLocalDataWrapper1472 &EagerContextThreadLocalDataWrapper::set_device_name) in PYBIND11_MODULE()