Searched refs:layer_func (Results 1 – 1 of 1) sorted by relevance
546 def add(self, layer_func): argument547 if isinstance(layer_func, base.Layer):548 args = function_utils.fn_args(layer_func.call)549 self.track_layer(layer_func)550 elif callable(layer_func):551 args = function_utils.fn_args(layer_func)555 "not '%s' of type '%s'." % (layer_func, type(layer_func)))556 self._layers_funcs.append((("training" in args), layer_func))