Searched refs:learning_rate_fn (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | feature_column.py | 60 learning_rate_fn=None, argument
151 learning_rate_fn=learning_rate_fn,
167 learning_rate_fn=None, argument
287 learning_rate_fn=learning_rate_fn,
300 learning_rate_fn=None): argument
303 self._learning_rate_fn = learning_rate_fn
374 learning_rate_fn=None, argument
403 learning_rate_fn=None, argument
410 learning_rate_fn=learning_rate_fn)
520 learning_rate_fn=None, argument
[all …]
|
| D | feature_column_v2.py | 60 learning_rate_fn=None, argument
195 learning_rate_fn=learning_rate_fn,
204 learning_rate_fn=learning_rate_fn,
217 learning_rate_fn=None, argument
401 learning_rate_fn=learning_rate_fn,
411 learning_rate_fn=learning_rate_fn,
429 learning_rate_fn=None, argument
459 learning_rate_fn=None, argument
466 learning_rate_fn=learning_rate_fn)
628 learning_rate_fn=None, argument
[all …]
|
| D | tpu_embedding.py | 73 learning_rate_fn=None, argument
133 if learning_rate is not None and learning_rate_fn is not None:
136 learning_rate, learning_rate_fn))
148 learning_rate_fn, optimization_parameters)
1368 set(c.learning_rate_fn
1370 if c.learning_rate_fn is not None))
1467 elif table_config.learning_rate_fn:
1469 self._learning_rate_fn_to_tag[table_config.learning_rate_fn])
|
| D | feature_column_v2_test.py | 243 learning_rate_fn=None,
270 learning_rate_fn=None,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.tpu.experimental.pbtxt | 37 …ension\', \'combiner\', \'initializer\', \'max_sequence_length\', \'learning_rate_fn\', \'embeddin…
45 …, \'shared_embedding_collection_name\', \'max_sequence_lengths\', \'learning_rate_fn\', \'embeddin…
|