| /external/tensorflow/tensorflow/python/distribute/ |
| shared_variable_creator.py | 43 `next_creator`, and stores it in the provided `shared_variable_store`.
66 def create_new_variable(next_creator, *args, **kwargs):
67 """Create the variable using `next_creator` and store it."""
69 v = next_creator(*args, **kwargs)
76 def reuse_variable(next_creator, *args, **kwargs):
78 del next_creator, args
|
| numpy_dataset.py | 78 def create_colocated_variable(next_creator, *args, **kwargs):
80 return next_creator(*args, **kwargs)
|
| one_device_strategy.py | 61 def _create_variable(self, next_creator, *args, **kwargs):
65 return next_creator(*args, **kwargs)
68 return next_creator(*args, **kwargs)
71 return next_creator(*args, **kwargs)
|
| distribute_lib.py | [all...] |
| collective_all_reduce_strategy.py | 237 def _create_variable(self, next_creator, *args, **kwargs):
244 return next_creator(*args, **kwargs)
316 v = next_creator(*args, **kwargs)
|
| parameter_server_strategy.py | 302 def _create_variable(self, next_creator, *args, **kwargs):
323 v = next_creator(*args, **kwargs)
332 # If "trainable" is True, next_creator() will add the contained
335 # to False for next_creator() since that causes functions like
348 var_creator = next_creator
|
| distribute_lib_test.py | 57 def _create_variable(self, next_creator, *args, **kwargs):
|
| mirrored_strategy.py | 265 # If "trainable" is True, next_creator() will add the member variables
268 # "trainable" to False for next_creator() since that causes functions
519 def _create_variable(self, next_creator, *args, **kwargs):
527 return next_creator(*args, **kwargs)
557 v = next_creator(*args, **kwargs)
[all...] |
| tpu_strategy.py | 110 # If "trainable" is True, next_creator() will add the member variables
113 # "trainable" to False for next_creator() since that causes functions
397 def _create_variable(self, next_creator, *args, **kwargs):
405 return next_creator(*args, **kwargs)
432 v = next_creator(*args, **kwargs)
|
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| agn_optimizer.py | 155 def _variable_creator(next_creator, collections, **kwargs):
162 return next_creator(collections=collections, **kwargs)
|
| /external/tensorflow/tensorflow/python/eager/ |
| wrap_function.py | 52 def variable_creator_scope(self, next_creator, **kwargs):
66 v = next_creator(**kwargs)
|
| def_function_test.py | 418 def capture_creator(next_creator, **kwargs):
419 created = next_creator(**kwargs)
|
| /external/tensorflow/tensorflow/python/training/ |
| checkpoint_management.py | 677 def _initializing_creator(next_creator, **kwargs):
679 v = next_creator(**kwargs)
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| minimize_loss_test.py | 138 def appending_creator(next_creator, *args, **kwargs):
139 v = next_creator(*args, **kwargs)
|
| mirrored_strategy_multigpu_test.py | 232 def thread_creator_fn(next_creator, *args, **kwargs):
233 return next_creator(*args, **kwargs) + ":thread_" + replica_id_str
243 def main_thread_creator(next_creator, *args, **kwargs):
244 # We are not using the underlying next_creator for test purposes.
245 del next_creator, args, kwargs
[all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| variable_scope_test.py | [all...] |
| /external/tensorflow/tensorflow/python/training/tracking/ |
| util.py | 393 def _trackable_custom_creator(next_creator, name, initial_value,
410 next_creator: See `variable_scope.variable_creator_scope`; the next
424 The output of `next_creator`: the fetched/created variable object.
429 return next_creator(
452 return next_creator(
[all...] |
| /external/tensorflow/tensorflow/python/keras/layers/ |
| core.py | 777 def _variable_creator(self, next_creator, **kwargs):
781 var = next_creator(**kwargs)
[all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| variable_scope.py | [all...] |
