Home
last modified time | relevance | path

Searched refs:caching_device (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dvariable_scope.py233 trainable=True, collections=None, caching_device=None, argument
352 trainable=True, collections=None, caching_device=None, argument
372 caching_device=caching_device,
390 caching_device=caching_device,
407 caching_device=caching_device, validate_shape=validate_shape,
423 "caching_device": caching_device,
437 caching_device=caching_device, partitioner=partitioner,
444 trainable=True, collections=None, caching_device=None, argument
665 caching_device=caching_device,
697 caching_device=None, argument
[all …]
Dresource_variable_ops.py210 caching_device=None, argument
279 caching_device=caching_device,
290 caching_device=None, argument
456 if caching_device is not None:
465 with ops.device(caching_device):
475 if caching_device:
476 with ops.device(caching_device):
Dfunctional_ops.py102 if varscope.caching_device is None:
190 if varscope.caching_device is None:
355 if varscope.caching_device is None:
548 if varscope.caching_device is None:
Dvariables.py140 caching_device=None, argument
230 caching_device=caching_device,
250 caching_device=None, argument
382 if caching_device is not None:
383 with ops.device(caching_device):
390 self._caching_device = caching_device
Drnn.py579 if varscope.caching_device is None:
1031 if varscope.caching_device is None:
1246 if varscope.caching_device is None:
/external/tensorflow/tensorflow/tools/api/golden/
Dtensorflow.-variable-scope.pbtxt6 name: "caching_device"
51 …=[\'self\', \'reuse\', \'name\', \'initializer\', \'regularizer\', \'caching_device\', \'partition…
59 …zer\', \'regularizer\', \'reuse\', \'trainable\', \'collections\', \'caching_device\', \'partition…
75 argspec: "args=[\'self\', \'caching_device\'], varargs=None, keywords=None, defaults=None"
Dtensorflow.variable_scope.pbtxt7 …', \'default_name\', \'values\', \'initializer\', \'regularizer\', \'caching_device\', \'partition…
Dtensorflow.-variable.pbtxt48 …itial_value\', \'trainable\', \'collections\', \'validate_shape\', \'caching_device\', \'name\', \…
Dtensorflow.pbtxt1113 … \'initializer\', \'regularizer\', \'trainable\', \'collections\', \'caching_device\', \'partition…
2085 …me_or_scope\', \'default_name\', \'initializer\', \'regularizer\', \'caching_device\', \'partition…
/external/tensorflow/tensorflow/contrib/framework/python/ops/
Dvariables.py198 caching_device=None, device=None, argument
244 caching_device=caching_device,
252 caching_device=None, device=None, partitioner=None, argument
289 caching_device=caching_device, device=device,
/external/tensorflow/tensorflow/contrib/model_pruning/python/layers/
Dlayers.py46 caching_device=None, argument
65 caching_device=caching_device,
/external/tensorflow/tensorflow/python/eager/
Dgraph_callable.py115 … trainable=True, collections=None, caching_device=None, # pylint: disable=redefined-outer-name argument
119 del collections, initializer, trainable, reuse, caching_device, shape,
141 … trainable=True, collections=None, caching_device=None, # pylint: disable=redefined-outer-name argument
144 del getter, regularizer, collections, caching_device, partitioner
/external/tensorflow/tensorflow/python/kernel_tests/
Dvariable_scope_test.py200 caching_device = "/job:moo"
203 "caching", caching_device=caching_device):
205 self.assertTrue(v.value().device.startswith(caching_device))
209 self.assertTrue(v2.value().device.startswith(caching_device))
211 with variable_scope.variable_scope("not_cached", caching_device=""):
214 caching_device))
218 caching_device=lambda op: op.device):
221 caching_device))
229 self.assertFalse(v_tower.value().device.startswith(caching_device))
Dresource_variable_ops_test.py390 2.0, caching_device="/job:localhost")
397 2.0, caching_device="/job:localhost")
478 caching_device="cpu:0",
Dvariables_test.py277 var_cached = variables.Variable(2.0, caching_device="/job:foo")
/external/tensorflow/tensorflow/contrib/seq2seq/python/ops/
Ddecoder.py192 if varscope.caching_device is None:
/external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/
Dmasked_autoregressive.py254 if vs.caching_device is None:
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dlayers.py1581 caching_device=None, argument
1600 caching_device=caching_device,
/external/tensorflow/tensorflow/python/training/
Dsaver_test.py250 v = resource_variable_ops.ResourceVariable([1], caching_device="/cpu:0",