Searched refs:resource_device (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | tensor_handle.h | 63 Device* resource_device, DataType dtype, EagerContext* ctx);
68 Device* op_device, Device* resource_device, EagerContext* ctx);
86 tensorflow::Device* resource_device() const { return resource_device_; } in resource_device() function
|
| D | tensor_handle.cc | 63 Device* resource_device, DataType dtype, in TensorHandle() argument
70 resource_device_(resource_device), in TensorHandle()
84 Device* op_device, Device* resource_device, in TensorHandle() argument
90 resource_device_(resource_device), in TensorHandle()
|
| D | execute.cc | 827 Device* resource_device = tensor_handle->resource_device(); in MaybeUpdateOpDevice() local
830 << " input device = " << resource_device->name() in MaybeUpdateOpDevice()
837 if (resource_device != op_device || op->Device() == nullptr) { in MaybeUpdateOpDevice()
838 VLOG(1) << (resource_device != op_device ? "Changing " : "Setting ") in MaybeUpdateOpDevice()
840 << resource_device->name() << " because input #" << i in MaybeUpdateOpDevice()
842 op->SetDevice(resource_device); in MaybeUpdateOpDevice()
|