Home
last modified time | relevance | path

Searched refs:mask_update_op (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/contrib/model_pruning/python/
Dpruning_test.py108 mask_update_op = p.mask_update_op()
112 session.run(mask_update_op)
179 mask_update_op = p.mask_update_op()
182 session.run(mask_update_op)
200 mask_update_op = p.conditional_mask_update_op()
208 session.run(mask_update_op)
234 mask_update_op = p.conditional_mask_update_op()
240 session.run(mask_update_op)
Dlearning.py63 mask_update_op, argument
162 sess.run(mask_update_op)
Dstrip_pruning_vars_test.py72 self.mask_update_op = None
128 self.mask_update_op = p.conditional_mask_update_op()
132 session.run(self.mask_update_op)
Dpruning.py519 def mask_update_op(self): member in Pruning
551 def mask_update_op(): function
552 return self.mask_update_op()
557 return control_flow_ops.cond(maybe_update_masks(), mask_update_op,
/external/tensorflow/tensorflow/contrib/model_pruning/examples/cifar10/
Dcifar10_train.py84 mask_update_op = pruning_obj.conditional_mask_update_op()
124 mon_sess.run(mask_update_op)
/external/tensorflow/tensorflow/contrib/model_pruning/
DREADME.md107 mask_update_op = p.conditional_mask_update_op()
116 # Update the masks by running the mask_update_op
117 mon_sess.run(mask_update_op)