Searched refs:group_by_size (Results 1 – 3 of 3) sorted by relevance
131 packs = cross_device_utils.group_by_size(values, bytes_per_pack=200)145 packs = cross_device_utils.group_by_size(values, bytes_per_pack=0)160 packs = cross_device_utils.group_by_size(values, bytes_per_pack=1)
1111 packs = cross_device_utils.group_by_size(flat_values,1231 packs = cross_device_utils.group_by_size(values_by_device[device_id],1243 packs = cross_device_utils.group_by_size(
658 def group_by_size(input_tensors, bytes_per_pack): function