Home
last modified time | relevance | path

Searched refs:handle_device (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dsession_ops.py215 handle_device = TensorHandle._get_device_name(handle)
216 with ops.device(handle_device):
240 handle_device = TensorHandle._get_device_name(handle)
241 with ops.device(handle_device):
261 handle_device = TensorHandle._get_device_name(handle)
262 with graph.as_default(), graph.device(handle_device):
276 handle_device = TensorHandle._get_device_name(handle)
277 if feeder.op.device == handle_device:
297 handle_device = TensorHandle._get_device_name(handle)
298 with graph.as_default(), graph.device(handle_device):
Dresource_variable_ops.py236 def __init__(self, handle, handle_device): argument
242 self._handle_device = handle_device
629 handle=self._handle, handle_device=self._handle.device)
Dsummary_ops_v2.py217 handle=self._resource, handle_device="cpu:0")
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dthreadpool.py55 handle=self._resource, handle_device=context.context().device_name)
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute.cc97 Device* handle_device = (*handle)->device(); in MaybeCopyInputToExpectedDevice() local
99 handle_device == nullptr ? ctx->HostCPU() : handle_device; in MaybeCopyInputToExpectedDevice()
148 int device_idx = StepStatsDeviceIndex(step_stats, ctx, handle_device); in MaybeCopyInputToExpectedDevice()
/external/tensorflow/tensorflow/python/data/ops/
Dmulti_device_iterator_ops.py242 handle_device=self._source_device)
Diterator_ops.py549 handle=self._iterator_resource, handle_device=self._device)
/external/tensorflow/tensorflow/python/eager/
Ddef_function.py209 handle=self._handle, handle_device=self._handle.device)