| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | tensor_handle.h | 65 Device* d, Device* op_device, EagerContext* ctx);
69 Device* op_device, EagerContext* ctx);
71 Device* d, Device* op_device, Device* resource_device,
88 Device* op_device, EagerContext* ctx,
90 static Status CreateEmptyLocalHandle(bool async, Device* d, Device* op_device,
121 Device* op_device() const { return op_device_; } in op_device() function
|
| D | tensor_handle.cc | 108 Device* op_device, EagerContext* ctx, in CreateLocalHandle() argument
112 t.dtype(), d, op_device, ctx); in CreateLocalHandle()
116 resource_handle, d, op_device, ctx); in CreateLocalHandle()
123 DataType dtype, Device* d, Device* op_device, in TensorHandle() argument
127 op_device_(op_device), in TensorHandle()
143 Device* op_device, EagerContext* ctx) in TensorHandle() argument
146 op_device_(op_device), in TensorHandle()
162 Device* op_device, in CreateEmptyLocalHandle() argument
167 d, op_device, resource_device, dtype, ctx); in CreateEmptyLocalHandle()
173 bool async, Device* d, Device* op_device, in TensorHandle() argument
[all …]
|
| D | execute.cc | 94 Device* op_device, in CopyInputToExpectedDevice() argument
102 const string& op_device_name = DeviceNameOrUnspecified(op_device); in CopyInputToExpectedDevice()
727 tensorflow::Device* op_device = op->Device(); in EagerRemoteExecute() local
742 id, i, remote_task, context_id, output_dtypes[i], op_device, &ctx, in EagerRemoteExecute()
773 std::move(request), op_device, eager_client.get(), in EagerRemoteExecute()
830 Device* op_device = op->Device() == nullptr ? ctx.HostCPU() : op->Device(); in MaybeUpdateOpDevice() local
838 << ", op device = " << op_device->name(); in MaybeUpdateOpDevice()
844 if (resource_device != op_device || op->Device() == nullptr) { in MaybeUpdateOpDevice()
845 DVLOG(1) << (resource_device != op_device ? "Changing " : "Setting ") in MaybeUpdateOpDevice()
861 << ", op device = " << op_device->name(); in MaybeUpdateOpDevice()
[all …]
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_mgr.cc | 127 out->set_op_device(in->op_device() ? in->op_device()->name() : ""); in SerializeRemoteTensorHandle()
147 if (parent_->local_device_mgr()->LookupDevice(in.op_device(), &device).ok() || in DeserializeRemoteTensorHandle()
155 in.op_device().empty() ? in.device() : in.op_device(); in DeserializeRemoteTensorHandle()
|
| /external/tensorflow/tensorflow/python/client/ |
| D | virtual_gpu_test.py | 104 op_device = ()
109 op_device += (device_index,)
111 result.append(op_device)
|
| /external/tensorflow/tensorflow/c/ |
| D | c_test_util.cc | 210 TF_Operation** op, const string& op_device, bool check) { in BinaryOpHelper() argument
212 if (!op_device.empty()) { in BinaryOpHelper()
213 TF_SetDevice(desc, op_device.c_str()); in BinaryOpHelper()
225 const string& op_device, TF_Status* s, in MinWithDevice() argument
228 BinaryOpHelper("Min", l, r, graph, s, name, &op, op_device, true); in MinWithDevice()
|
| D | c_test_util.h | 93 const string& op_device, TF_Status* s,
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | remote_tensor_handle.proto | 30 string op_device = 4; field
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 1012 tensorflow::Device* d = handle_->op_device(); in DeviceName()
|