Searched refs:get_train_step (Results 1 – 7 of 7) sorted by relevance
10 name: "get_train_step"
60 train_step = model.get_train_step(loss)
105 def get_train_step(self, loss): member in _ComposableModel
263 sdca_model, train_op = optimizer.get_train_step(columns_to_variables,
118 def get_train_step(self, columns_to_variables, weight_column_name, loss_type, member in SDCAOptimizer
161 sdca_model, train_op = optimizer.get_train_step(