| /external/tensorflow/tensorflow/python/distribute/ |
| D | shared_variable_creator.py | 66 def create_new_variable(next_creator, **kwargs): argument
69 v = next_creator(**kwargs)
76 def reuse_variable(next_creator, **kwargs): argument
78 del next_creator
|
| D | parameter_server_strategy_v2.py | 527 def _create_variable(self, next_creator, **kwargs): argument
553 var = next_creator(**kwargs)
560 return self._create_variable_round_robin(next_creator, **kwargs)
595 return self._create_variable_round_robin(next_creator, **kwargs)
605 return self._create_variable_round_robin(next_creator, **kwargs)
668 var_list.append(self._create_variable_round_robin(next_creator, **kwargs))
673 def _create_variable_round_robin(self, next_creator, **kwargs): argument
679 var = next_creator(**kwargs)
|
| D | one_device_strategy.py | 274 def _create_variable(self, next_creator, **kwargs): argument
278 return next_creator(**kwargs)
281 return next_creator(**kwargs)
284 return next_creator(**kwargs)
|
| D | numpy_dataset.py | 79 def create_colocated_variable(next_creator, **kwargs): argument
81 return next_creator(**kwargs)
|
| D | distribute_lib_test.py | 79 def _create_variable(self, next_creator, **kwargs): argument
211 def creator(next_creator, **kwargs): argument
212 return next_creator(**kwargs)
436 def _create_variable(self, next_creator, **kwargs): argument
437 return next_creator(**kwargs)
|
| D | distribute_lib.py | 2071 def creator_with_resource_vars(next_creator, **kwargs): argument
2091 created = self._create_variable(next_creator, **kwargs)
2123 def _create_variable(self, next_creator, **kwargs): argument
2190 def create_colocated_variable(next_creator, **kwargs): argument
2194 return next_creator(**kwargs)
3376 def creator(next_creator, **kwargs): argument
3378 return next_creator(**kwargs)
|
| D | parameter_server_strategy.py | 427 def _create_variable(self, next_creator, **kwargs): argument
448 v = next_creator(**kwargs)
473 var_creator = next_creator
|
| D | mirrored_strategy.py | 492 def _create_variable(self, next_creator, **kwargs): argument
499 return next_creator(**kwargs)
523 v = next_creator(**kwargs)
|
| D | tpu_strategy.py | 1120 def _create_variable(self, next_creator, **kwargs): argument
1123 return next_creator(**kwargs)
1130 return next_creator(**kwargs)
1157 v = next_creator(**kwargs)
|
| D | mirrored_strategy_test.py | 341 def thread_creator_fn(next_creator, **kwargs): argument
342 return next_creator(**kwargs) + ":thread_" + replica_id_str
352 def main_thread_creator(next_creator, **kwargs): argument
354 del next_creator, kwargs
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | sharded_variable_test.py | 77 def sharded_variable_creator(next_creator, **kwargs): argument
83 v1 = next_creator(**kwargs)
87 v2 = next_creator(**kwargs)
|
| D | minimize_loss_test.py | 178 def appending_creator(next_creator, **kwargs): argument
179 v = next_creator(**kwargs)
|
| /external/tensorflow/tensorflow/python/distribute/parallel_device/ |
| D | saving.py | 104 def _variable_creator(next_creator, parallel_device, **kwargs): argument
111 variable = next_creator(**kwargs)
|
| /external/tensorflow/tensorflow/python/training/ |
| D | checkpoint_management.py | 798 def _initializing_creator(next_creator, **kwargs): argument
800 v = next_creator(**kwargs)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | variable_scope_test.py | 1701 def creator_a(next_creator, **kwargs): argument
1703 return next_creator(**kwargs)
1705 def creator_b(next_creator, **kwargs): argument
1707 return next_creator(**kwargs)
1717 def creater_c(next_creator, **kwargs): argument
1723 return next_creator(**kwargs)
1736 def creator(next_creator, **kwargs): argument
1737 return next_creator(**kwargs)
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | wrap_function.py | 59 def variable_creator_scope(self, next_creator, **kwargs): argument
74 v = next_creator(**kwargs)
|
| D | def_function_test.py | 504 def capture_creator(next_creator, **kwargs): argument
505 created = next_creator(**kwargs)
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | core.py | 911 def _variable_creator(next_creator, **kwargs): argument
912 var = next_creator(**kwargs)
1348 def _variable_creator(next_creator, **creator_kwargs): argument
1349 var = next_creator(**creator_kwargs)
|
| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | util.py | 564 def _trackable_custom_creator(next_creator, argument
604 return next_creator(initial_value=initializer, name=name, **inner_kwargs)
625 return next_creator(
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | load.py | 601 def uninitialized_variable_creator(next_creator, **kwargs): argument
603 del next_creator
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | variable_scope.py | 2584 def default_variable_creator(next_creator=None, **kwargs): argument
2586 assert next_creator is None
2643 def default_variable_creator_v2(next_creator=None, **kwargs): argument
2645 assert next_creator is None
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding_v2.py | 1711 next_creator: Callable[..., tf_variables.Variable], *args, **kwargs):
1752 variables.append(next_creator(*args, **kwargs))
|
| /external/tensorflow/tensorflow/python/distribute/coordinator/ |
| D | cluster_coordinator.py | 1316 def disallow_variable_creation(next_creator, **kwargs): argument
|