Searched refs:init_kwargs (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/framework/ |
| D | device_spec.py | 257 init_kwargs = dict(
262 init_kwargs.update(kwargs)
263 return self.__class__(**init_kwargs)
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | optimizer_v2_test.py | 1180 def test_duplicate_ops(self, optimizer_class, init_kwargs=None): argument
1181 init_kwargs = init_kwargs or {}
1182 optimizer = optimizer_class(**init_kwargs)
1205 def test_subclass_compat(self, optimizer_class, init_kwargs=None): argument
1217 init_kwargs = init_kwargs or {}
1218 optimizer = SubclassedOptimizer(**init_kwargs)
|
| /external/tensorflow/tensorflow/python/keras/saving/saved_model/ |
| D | saved_model_test.py | 1072 def test_metric(self, metric_cls, num_tensor_args, shape, init_kwargs=None): argument
1073 init_kwargs = init_kwargs or {}
1074 metric = metric_cls(**init_kwargs)
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | recurrent.py | 1193 init_kwargs = dict(inputs=inputs, training=training, count=count)
1194 return self._dropout_mask_cache.setdefault(kwargs=init_kwargs)
1214 init_kwargs = dict(inputs=inputs, training=training, count=count)
1215 return self._recurrent_dropout_mask_cache.setdefault(kwargs=init_kwargs)
|