Searched refs:handle_device (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | session_ops.py | 214 handle_device = TensorHandle._get_device_name(handle)
215 with ops.device(handle_device):
239 handle_device = TensorHandle._get_device_name(handle)
240 with ops.device(handle_device):
260 handle_device = TensorHandle._get_device_name(handle)
261 with graph.as_default(), graph.device(handle_device):
275 handle_device = TensorHandle._get_device_name(handle)
276 if feeder.op.device == handle_device:
296 handle_device = TensorHandle._get_device_name(handle)
297 with graph.as_default(), graph.device(handle_device):
|
| D | resource_variable_ops.py | 265 def __init__(self, handle, handle_device): argument
271 self._handle_device = handle_device
436 handle=self._handle, handle_device=self._handle.device)
|
| D | summary_ops_v2.py | 295 handle=self._resource, handle_device="cpu:0")
|
| D | data_flow_ops.py | 1354 handle=accumulator_ref, handle_device=context.context().device_name)
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute.cc | 107 int i, Device* handle_device, in CopyInputToExpectedDevice() argument
111 DCHECK(expected_input_device != handle_device); in CopyInputToExpectedDevice()
136 handle_device->name(), " (operation running on ", op_device_name, ")", in CopyInputToExpectedDevice()
145 << " but is actually on " << handle_device->name() in CopyInputToExpectedDevice()
158 return absl::StrCat("_Send input ", i, " from ", handle_device->name(), in CopyInputToExpectedDevice()
169 absl::StrCat("Failed copying input tensor from ", handle_device->name(), in CopyInputToExpectedDevice()
216 Device* handle_device = handle->DeviceOrHostCPU(*ctx); in ValidateInputTypeAndPlacement() local
220 if (expected_device != handle_device && maybe_copy) { in ValidateInputTypeAndPlacement()
222 handle, i, handle_device, in ValidateInputTypeAndPlacement()
875 Device* handle_device = handle->DeviceOrHostCPU(ctx); in EagerRemoteExecute() local
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | threadpool.py | 55 handle=self._resource, handle_device=context.context().device_name)
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | multi_device_iterator_ops.py | 262 handle_device=self._source_device)
|