Searched refs:lr_function (Results 1 – 2 of 2) sorted by relevance
62 def lr_function(): function68 optimizer = tpu_embedding_v2_utils.SGD(learning_rate=lr_function)
935 void lr_function() __attribute__((lock_returned)); // \