Home
last modified time | relevance | path

Searched refs:WEIGHT_COLLECTION (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/contrib/model_pruning/python/layers/
Dcore_layers.py36 WEIGHT_COLLECTION = 'kernel' variable
163 ops.add_to_collection(WEIGHT_COLLECTION, self.kernel)
437 ops.add_to_collection(WEIGHT_COLLECTION, self.kernel)
Drnn_cells.py120 ops.add_to_collection(core_layers.WEIGHT_COLLECTION, self._kernel)
269 ops.add_to_collection(core_layers.WEIGHT_COLLECTION, self._kernel)
/external/tensorflow/tensorflow/contrib/model_pruning/python/
Dpruning.py84 _WEIGHT_COLLECTION = core.WEIGHT_COLLECTION