Home
last modified time | relevance | path

Searched refs:MASKED_WEIGHT_COLLECTION (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/contrib/model_pruning/python/layers/
Dlayers_test.py55 masked_weight = ops.get_collection(core_layers.MASKED_WEIGHT_COLLECTION)
83 masked_weight = ops.get_collection(core_layers.MASKED_WEIGHT_COLLECTION)
104 masked_weight = ops.get_collection(core_layers.MASKED_WEIGHT_COLLECTION)
130 masked_weight = ops.get_collection(core_layers.MASKED_WEIGHT_COLLECTION)
Dcore_layers.py35 MASKED_WEIGHT_COLLECTION = 'masked_weights' variable
161 ops.add_to_collection(MASKED_WEIGHT_COLLECTION, self.masked_kernel)
435 ops.add_to_collection(MASKED_WEIGHT_COLLECTION, self.masked_kernel)
Drnn_cells.py117 ops.add_to_collection(core_layers.MASKED_WEIGHT_COLLECTION,
266 ops.add_to_collection(core_layers.MASKED_WEIGHT_COLLECTION,
/external/tensorflow/tensorflow/contrib/model_pruning/python/
Dpruning.py83 _MASKED_WEIGHT_COLLECTION = core.MASKED_WEIGHT_COLLECTION