Searched defs:add_slot (Results 1 – 2 of 2) sorted by relevance
861 def add_slot(self, var, slot_name, initializer="zeros", shape=None): member in OptimizerV2
879 def add_slot(self, var, slot_name, initializer='zeros'): member in LossScaleOptimizer