| /external/tensorflow/tensorflow/python/ops/ |
| D | variable_scope.py | 292 caching_device=None, argument
444 caching_device=None, argument
469 caching_device=caching_device,
490 caching_device=caching_device,
514 caching_device=caching_device,
538 "caching_device": caching_device,
561 caching_device=caching_device,
579 caching_device=None, argument
778 caching_device=caching_device,
810 caching_device=None, argument
[all …]
|
| D | variables.py | 143 caching_device=None, argument
166 caching_device=caching_device,
181 caching_device=None, argument
201 caching_device=caching_device,
341 caching_device=None, argument
1276 caching_device=None, argument
1360 caching_device=None, argument
1439 caching_device=caching_device,
1459 caching_device=None, argument
1601 if caching_device is not None:
[all …]
|
| D | resource_variable_ops.py | 348 caching_device=None, argument
419 caching_device=caching_device,
437 caching_device=None, argument
589 if caching_device is not None:
598 with ops.device(caching_device):
608 if caching_device:
609 with ops.device(caching_device):
|
| D | functional_ops.py | 113 if varscope.caching_device is None:
222 if varscope.caching_device is None:
389 if varscope.caching_device is None:
|
| D | map_fn.py | 198 if varscope.caching_device is None:
|
| D | rnn.py | 611 if varscope.caching_device is None:
1083 if varscope.caching_device is None:
1300 if varscope.caching_device is None:
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-variable-scope.pbtxt | 6 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"
|
| D | tensorflow.variable_scope.pbtxt | 7 …', \'default_name\', \'values\', \'initializer\', \'regularizer\', \'caching_device\', \'partition…
|
| D | tensorflow.-variable.pbtxt | 53 …itial_value\', \'trainable\', \'collections\', \'validate_shape\', \'caching_device\', \'name\', \…
|
| D | tensorflow.pbtxt | 1313 … \'initializer\', \'regularizer\', \'trainable\', \'collections\', \'caching_device\', \'partition…
1337 … \'initializer\', \'regularizer\', \'trainable\', \'collections\', \'caching_device\', \'partition…
2405 …me_or_scope\', \'default_name\', \'initializer\', \'regularizer\', \'caching_device\', \'partition…
|
| /external/tensorflow/tensorflow/contrib/framework/python/ops/ |
| D | variables.py | 210 caching_device=None, argument
273 caching_device=caching_device,
288 caching_device=None, argument
344 caching_device=caching_device,
|
| /external/tensorflow/tensorflow/contrib/model_pruning/python/layers/ |
| D | layers.py | 46 caching_device=None, argument
65 caching_device=caching_device,
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | base_layer_utils.py | 68 caching_device=None, argument
145 caching_device=caching_device,
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | def_function.py | 53 caching_device=None, argument
104 caching_device=caching_device, name=name, dtype=dtype,
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | variable_scope_test.py | 318 caching_device = "/job:moo"
321 "caching", caching_device=caching_device):
323 self.assertTrue(v.value().device.startswith(caching_device))
327 self.assertTrue(v2.value().device.startswith(caching_device))
329 with variable_scope.variable_scope("not_cached", caching_device=""):
332 v2_not_cached.value().device.startswith(caching_device))
336 caching_device=lambda op: op.device):
339 v2_identity_device.value().device.startswith(caching_device))
347 self.assertFalse(v_tower.value().device.startswith(caching_device))
|
| D | resource_variable_ops_test.py | 178 v = resource_variable_ops.ResourceVariable(0.0, caching_device="cpu:0")
639 caching_device="cpu:0",
796 2.0, caching_device="/job:localhost")
803 2.0, caching_device="/job:localhost")
888 caching_device="cpu:0",
|
| D | variables_test.py | 297 var_cached = variables.Variable(2.0, caching_device="/job:foo")
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| D | rnn.py | 202 if varscope.caching_device is None:
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.-variable.pbtxt | 52 …s=[\'self\', \'initial_value\', \'trainable\', \'validate_shape\', \'caching_device\', \'name\', \…
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_map_ops.py | 208 if varscope.caching_device is None:
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/ |
| D | masked_autoregressive.py | 276 if vs.caching_device is None:
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/ops/ |
| D | decoder.py | 324 if varscope.caching_device is None:
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | layers.py | 1715 caching_device=None, argument
1736 caching_device=caching_device,
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu.py | 1561 prev_caching_device = vscope.caching_device
|
| /external/tensorflow/tensorflow/python/training/ |
| D | saver_test.py | 273 v = resource_variable_ops.ResourceVariable([1], caching_device="/cpu:0",
|