Home
last modified time | relevance | path

Searched refs:set_caching_device (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dfunctional_ops.py116 varscope.set_caching_device(lambda op: op.device)
150 varscope.set_caching_device(None)
225 varscope.set_caching_device(lambda op: op.device)
261 varscope.set_caching_device(None)
392 varscope.set_caching_device(lambda op: op.device)
491 varscope.set_caching_device(None)
Dmap_fn.py201 varscope.set_caching_device(lambda op: op.device)
283 varscope.set_caching_device(None)
Drnn.py612 varscope.set_caching_device(lambda op: op.device)
1084 varscope.set_caching_device(lambda op: op.device)
1301 varscope.set_caching_device(lambda op: op.device)
Dvariable_scope.py1141 def set_caching_device(self, caching_device): member in VariableScope
1862 variable_scope_object.set_caching_device(self._caching_device)
1921 variable_scope_object.set_caching_device(self._caching_device)
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_map_ops.py211 varscope.set_caching_device(lambda op: op.device)
308 varscope.set_caching_device(None)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-variable-scope.pbtxt74 name: "set_caching_device"
/external/tensorflow/tensorflow/lite/experimental/examples/lstm/
Drnn.py203 varscope.set_caching_device(lambda op: op.device)
/external/tensorflow/tensorflow/contrib/seq2seq/python/ops/
Ddecoder.py325 varscope.set_caching_device(lambda op: op.device)
/external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/
Dmasked_autoregressive.py277 vs.set_caching_device(lambda op: op.device)
/external/tensorflow/tensorflow/python/tpu/
Dtpu.py1563 vscope.set_caching_device(lambda op: op.device)
1568 vscope.set_caching_device(prev_caching_device)
/external/tensorflow/tensorflow/python/kernel_tests/
Dvariable_scope_test.py342 vs_live.set_caching_device("/job:live")