Searched refs:bucket_sizes_tensor_device (Results 1 – 1 of 1) sorted by relevance
1770 auto bucket_sizes_tensor_device = at::empty({(int64_t)num_buckets}, options); in sync_bucket_indices() local1771 bucket_sizes_tensor_device.copy_(bucket_sizes_tensor, /*non_blocking=*/true); in sync_bucket_indices()1773 bucket_sizes_tensor_device}; in sync_bucket_indices()