Searched defs:embedding_table_before (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/tests/ |
| H A D | tpu_embedding_v1_correctness_test.py | 196 def _check_embedding_and_slot_variables_for_sgd(self, embedding_table_before, argument
206 embedding_table_before, argument
221 def _check_embedding_and_slot_variables_for_adam(self, embedding_table_before, argument
242 def _check_embedding_and_slot_variables_for_ftrl(self, embedding_table_before, argument
|
| H A D | tpu_embedding_base_test.py | 480 def _check_embedding_and_slot_variables_for_sgd(self, embedding_table_before, argument
489 embedding_table_before, argument
504 self, embedding_table_before, gradients, optimizer, variable): argument
533 def _check_embedding_and_slot_variables_for_adam(self, embedding_table_before, argument
554 def _check_embedding_and_slot_variables_for_ftrl(self, embedding_table_before, argument
|