Searched refs:fill_empty_tensor_to_shards (Results 1 – 2 of 2) sorted by relevance
11 fill_empty_tensor_to_shards,100 fill_empty_tensor_to_shards(tensor_list, self.dim, num_empty_tensors),116 tensor_list = fill_empty_tensor_to_shards(
188 def fill_empty_tensor_to_shards( function