HomeSort by relevance Sort by last modified time
    Searched refs:_set_device (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/contrib/copy_graph/python/util/
copy_elements.py 224 new_op._set_device(device_function(new_op))
  /external/tensorflow/tensorflow/python/framework/
importer.py 338 op._set_device(coloc_device) # pylint: disable=protected-access
756 op._set_device(coloc_device) # pylint: disable=protected-access
ops.py 1855 def _set_device(self, device): # pylint: disable=redefined-outer-name member in class:Operation
    [all...]
ops_test.py 264 op._set_device("/job:goo/device:GPU:0")
268 op._set_device(
    [all...]
  /external/tensorflow/tensorflow/python/client/
session_test.py     [all...]
  /external/tensorflow/tensorflow/python/keras/_impl/keras/
backend.py 435 def _set_device(self, device): member in class:_TfDeviceCaptureOp
    [all...]

Completed in 292 milliseconds