Searched refs:conditional_mask_update_op (Results 1 – 5 of 5) sorted by relevance
200 mask_update_op = p.conditional_mask_update_op()234 mask_update_op = p.conditional_mask_update_op()
128 self.mask_update_op = p.conditional_mask_update_op()
533 def conditional_mask_update_op(self): member in Pruning
84 mask_update_op = pruning_obj.conditional_mask_update_op()
107 mask_update_op = p.conditional_mask_update_op()