| /external/tensorflow/tensorflow/python/ops/ |
| D | variable_scope.py | 325 caching_device=None, argument
474 caching_device=None, argument
499 caching_device=caching_device,
520 caching_device=caching_device,
544 caching_device=caching_device,
568 "caching_device": caching_device,
591 caching_device=caching_device,
609 caching_device=None, argument
802 caching_device=caching_device,
836 caching_device=None, argument
[all …]
|
| D | variables.py | 191 caching_device=None, argument
215 caching_device=caching_device,
231 caching_device=None, argument
252 caching_device=caching_device,
376 caching_device=None, argument
1459 caching_device=None, argument
1557 caching_device=None, argument
1653 caching_device=caching_device,
1676 caching_device=None, argument
1850 if caching_device is not None:
[all …]
|
| D | resource_variable_ops.py | 357 caching_device=None, argument
429 self._caching_device = caching_device
1573 caching_device=None, argument
1667 caching_device=caching_device,
1683 caching_device=None, argument
1876 if caching_device is not None:
1885 with ops.device(caching_device):
1894 if caching_device:
1895 with ops.device(caching_device):
1932 caching_device=caching_device,
[all …]
|
| D | functional_ops.py | 119 if varscope.caching_device is None:
315 if varscope.caching_device is None:
574 if varscope.caching_device is None:
|
| D | rnn.py | 689 if varscope.caching_device is None:
1161 if varscope.caching_device is None:
1387 if varscope.caching_device is None:
|
| D | map_fn.py | 411 if varscope.caching_device is None:
|
| /external/tensorflow/tensorflow/python/keras/legacy_tf_layers/ |
| D | variable_scope_shim.py | 163 caching_device=None, argument
306 caching_device=None, argument
334 caching_device=caching_device,
357 "caching_device": caching_device,
380 caching_device=caching_device,
398 caching_device=None, argument
502 caching_device=caching_device,
|
| /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 | 61 …itial_value\', \'trainable\', \'collections\', \'validate_shape\', \'caching_device\', \'name\', \…
|
| D | tensorflow.pbtxt | 1381 … \'initializer\', \'regularizer\', \'trainable\', \'collections\', \'caching_device\', \'partition…
1405 … \'initializer\', \'regularizer\', \'trainable\', \'collections\', \'caching_device\', \'partition…
2517 …me_or_scope\', \'default_name\', \'initializer\', \'regularizer\', \'caching_device\', \'partition…
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | def_function.py | 211 caching_device=None, argument
279 caching_device=caching_device, name=name, dtype=dtype,
314 caching_device=caching_device,
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | base_layer_utils.py | 56 caching_device=None, argument
131 caching_device=caching_device,
|
| D | base_layer_v1.py | 383 caching_device = kwargs.pop('caching_device', None)
436 if caching_device is not None:
440 caching_device = None
459 caching_device=caching_device)
|
| D | base_layer.py | 593 caching_device = kwargs.pop('caching_device', None)
647 if caching_device is not None:
651 caching_device = None
669 caching_device=caching_device)
|
| /external/tensorflow/tensorflow/python/kernel_tests/variables/ |
| D | variable_scope_test.py | 341 caching_device = "/job:moo"
344 "caching", caching_device=caching_device):
346 self.assertTrue(v.value().device.startswith(caching_device))
350 self.assertTrue(v2.value().device.startswith(caching_device))
352 with variable_scope.variable_scope("not_cached", caching_device=""):
355 v2_not_cached.value().device.startswith(caching_device))
359 caching_device=lambda op: op.device):
362 v2_identity_device.value().device.startswith(caching_device))
370 self.assertFalse(v_tower.value().device.startswith(caching_device))
|
| D | resource_variable_ops_test.py | 226 v = resource_variable_ops.ResourceVariable(0.0, caching_device="cpu:0")
882 caching_device="cpu:0",
1126 2.0, caching_device="/job:localhost")
1133 2.0, caching_device="/job:localhost")
1225 caching_device="cpu:0",
|
| D | variables_test.py | 332 var_cached = variables.Variable(2.0, caching_device="/job:foo")
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | recurrent.py | 1356 caching_device=default_caching_device)
1363 caching_device=default_caching_device)
1371 caching_device=default_caching_device)
1807 caching_device=default_caching_device)
1814 caching_device=default_caching_device)
1830 caching_device=default_caching_device)
2369 caching_device=default_caching_device)
2376 caching_device=default_caching_device)
2395 caching_device=default_caching_device)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.-variable.pbtxt | 60 …s=[\'self\', \'initial_value\', \'trainable\', \'validate_shape\', \'caching_device\', \'name\', \…
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | values_v2.py | 89 caching_device=None,
|
| D | packed_distributed_variable.py | 76 caching_device=None,
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu.py | 2299 prev_caching_device = vscope.caching_device
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | save_test.py | 499 obj.v = variables.Variable(2., caching_device=lambda op: op.device)
|
| D | load_test.py | 2183 obj.v = variables.Variable(2., caching_device=lambda op: op.device)
|