Searched defs:track_layer (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/contrib/eager/python/
H A Dnetwork.py60 `Layer`s (including other `Network`s) should be added via `track_layer`. They
68 self.layer_one = self.track_layer(tf.layers.Dense(16, input_shape=(8,)))
69 self.layer_two = self.track_layer(tf.layers.Dense(1, input_shape=(16,)))
111 # - Detect layers used in __call__ that weren't registered with track_layer.
302 "another Network via track_layer, or when they are first "
306 def track_layer(self, layer): member in class:Network
327 "Network.track_layer() passed type %s, not a tf.layers.Layer" %
503 self.track_layer(layer_func)
H A Dcheckpointable_utils_test.py69 self.track_layer(value, name=name)
74 def track_layer(self, layer, name): member in class:CheckpointableNetwork
76 return super(CheckpointableNetwork, self).track_layer(layer)
127 self._via_track_layer = self.track_layer(

Completed in 168 milliseconds