Calls and returns the retrieve ops for each embedding table. Returns: A list of ops to retrieve embedding and slot variables from TPU to CPU.
()
| 828 | return load_ops_list |
| 829 | |
| 830 | def retrieve_ops(): |
| 831 | """Calls and returns the retrieve ops for each embedding table. |
| 832 | |
| 833 | Returns: |
| 834 | A list of ops to retrieve embedding and slot variables from TPU to CPU. |
| 835 | """ |
| 836 | retrieve_ops_list = [] |
| 837 | for retrieve_op_fn in retrieve_op_fns: |
| 838 | retrieve_ops_list.extend(retrieve_op_fn()) |
| 839 | return retrieve_ops_list |
| 840 | |
| 841 | return VariablesAndOps(embedding_variables_by_table, |
| 842 | slot_variables_by_table, |