Searched defs:track_layer (Results 1 - 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/contrib/eager/python/ |
H A D | network.py | 60 `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 D | checkpointable_utils_test.py | 69 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