| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | tensor_handle.h | 60 TensorHandle(const Tensor& t, Device* d, Device* op_device,
62 TensorHandle(uint64 node_id, Device* d, Device* op_device,
68 Device* op_device, Device* resource_device, EagerContext* ctx);
85 tensorflow::Device* op_device() const { return op_device_; } in op_device() function
90 tensorflow::Device** op_device);
|
| D | tensor_handle.cc | 48 TensorHandle::TensorHandle(const class Tensor& t, Device* d, Device* op_device, in TensorHandle() argument
54 op_device_(op_device), in TensorHandle()
62 TensorHandle::TensorHandle(uint64 node_id, Device* d, Device* op_device, in TensorHandle() argument
69 op_device_(op_device), in TensorHandle()
84 Device* op_device, Device* resource_device, in TensorHandle() argument
89 op_device_(op_device), in TensorHandle()
160 tensorflow::Device** op_device) { in TensorAndDevice() argument
170 *op_device = op_device_; in TensorAndDevice()
|
| D | execute.cc | 294 bool IsMultiDevice(const FunctionDef* fdef, const string& op_device) { in IsMultiDevice() argument
705 tensorflow::Device* op_device = op->Device(); in EagerRemoteExecute()
731 /*d=*/op_device, /*op_device=*/op_device, in EagerRemoteExecute()
732 /*resource_device=*/output_dtypes[i] == DT_RESOURCE ? op_device in EagerRemoteExecute()
823 Device* op_device = op->Device() == nullptr ? ctx->HostCPU() : op->Device(); in MaybeUpdateOpDevice() local
831 << ", op device = " << op_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()
855 << ", op device = " << op_device->name(); in MaybeUpdateOpDevice()
1024 DCHECK_EQ(kernel->device(), retvals[i]->op_device()); in EagerKernelExecute()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_internal.h | 80 tensorflow::Device* op_device) in TFE_TensorHandle()
81 : handle(new tensorflow::TensorHandle(t, d, op_device, nullptr)) {} in TFE_TensorHandle()
|
| D | c_api.cc | 484 tensorflow::Device* d = h->handle->op_device(); in TFE_TensorHandleDeviceName()
524 tensorflow::Device* op_device = nullptr; in TFE_TensorHandleResolve() local
533 status->status = h_cpu->TensorAndDevice(&t, &d, &op_device); in TFE_TensorHandleResolve()
539 status->status = h->handle->TensorAndDevice(&t, &d, &op_device); in TFE_TensorHandleResolve()
548 status->status = h_cpu->TensorAndDevice(&t, &d, &op_device); in TFE_TensorHandleResolve()
884 tensorflow::Device* op_device = nullptr; in TFE_TensorHandleUnderlyingTensorInHostMemory() local
886 status->status = h->handle->TensorAndDevice(&t, &d, &op_device); in TFE_TensorHandleUnderlyingTensorInHostMemory()
|
| /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,
|