Searched refs:resource_device (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | placement_utils.cc | 154 Device* resource_device = tensor_handle->resource_device(); in MaybePinToResourceDevice() local
157 << " input device = " << resource_device->name() in MaybePinToResourceDevice()
164 if (resource_device != op_device || op.Device() == kVariantDeviceNull) { in MaybePinToResourceDevice()
165 DVLOG(1) << (resource_device != op_device ? "Changing " : "Setting ") in MaybePinToResourceDevice()
167 << resource_device->name() << " because input #" << i in MaybePinToResourceDevice()
169 *device = resource_device; in MaybePinToResourceDevice()
|
| D | tensor_handle.h | 59 Device* resource_device, EagerContext* ctx);
63 TensorHandle(Device* d, Device* op_device, Device* resource_device,
81 Device* resource_device,
84 Device* resource_device,
152 Device* resource_device() const { return resource_device_; } in resource_device() function
|
| D | tensor_handle.cc | 233 Device* resource_device, in CreateLocalHandle() argument
238 return new TensorHandle(std::move(t), d, op_device, resource_device, ctx); in CreateLocalHandle()
243 Device* resource_device, EagerContext* ctx) in TensorHandle() argument
248 resource_device_(resource_device), in TensorHandle()
279 Device* resource_device, in CreateEmptyLocalHandle() argument
282 return new TensorHandle(d, op_device, resource_device, dtype, ctx); in CreateEmptyLocalHandle()
286 Device* resource_device, tensorflow::DataType dtype, in TensorHandle() argument
292 resource_device_(resource_device), in TensorHandle()
|
| D | tensor_handle_test.cc | 271 th->resource_device()->attributes().incarnation())); in TEST()
277 th->resource_device()->attributes().incarnation())); in TEST()
|
| D | execute.cc | 1189 d, dstd, h->resource_device(), h->dtype, ctx); in LocalEagerCopyToDevice()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_execute_node.h | 74 if (input->resource_device() != nullptr && in RemoteExecuteNode()
75 input->resource_device() != device_) { in RemoteExecuteNode()
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
| D | saved_variable_loading_test.cc | 90 EXPECT_EQ(down_cast<TensorHandle*>(var->handle())->resource_device()->name(), in TEST_P()
|