Searched refs:keep_mask (Results 1 – 3 of 3) sorted by relevance
191 keep_mask = array_ops.tile(194 final_output = keep_mask * tf_output
1818 static void drop_capbset(uint64_t keep_mask, unsigned int last_valid_cap) in drop_capbset() argument1822 for (i = 0; i < sizeof(keep_mask) * 8 && i <= last_valid_cap; ++i) { in drop_capbset()1823 if (keep_mask & (one << i)) in drop_capbset()
4168 keep_mask = random_tensor >= rate4169 ret = x * scale * math_ops.cast(keep_mask, x.dtype)