Searched refs:expand_tensor (Results 1 – 3 of 3) sorted by relevance
76 expanded_tensor = pruning_utils.expand_tensor(tensor, block_dim)
96 def expand_tensor(tensor, block_dims): function
480 updated_mask = pruning_utils.expand_tensor(new_mask, self._block_dim)