Searched refs:DeviceOrHostCPU (Results 1 – 5 of 5) sorted by relevance
83 send_device_(src->DeviceOrHostCPU(*ctx)), in RemoteCopyNode()154 src_->DeviceOrHostCPU(*ctx_)->name()); in StartSend()314 op->set_device_name(packed_handle->DeviceOrHostCPU(*ctx)->name()); in SerializePackedHandle()345 h->DeviceOrHostCPU(*ctx)->name(), in SerializePackedHandle()
108 input_devices.push_back(h->DeviceOrHostCPU(*ctx)); in TEST()
216 Device* handle_device = handle->DeviceOrHostCPU(*ctx); in ValidateInputTypeAndPlacement()857 tensorflow::Device* input_device_or_cpu = input->DeviceOrHostCPU(ctx); in EagerRemoteExecute()875 Device* handle_device = handle->DeviceOrHostCPU(ctx); in EagerRemoteExecute()1219 auto send_device = h->DeviceOrHostCPU(*ctx); in EagerCopyToDevice()
161 Device* DeviceOrHostCPU(const EagerContext& ctx) const;
529 Device* TensorHandle::DeviceOrHostCPU(const EagerContext& ctx) const { in DeviceOrHostCPU() function in tensorflow::TensorHandle976 tensorflow::Device* srcd = DeviceOrHostCPU(ctx); in CopyToDevice()