Searched defs:get_train_step (Results 1 – 2 of 2) sorted by relevance
118 def get_train_step(self, columns_to_variables, weight_column_name, loss_type, member in SDCAOptimizer
105 def get_train_step(self, loss): member in _ComposableModel