Searched defs:variables_for_bucket (Results 1 – 1 of 1) sorted by relevance
492 auto variables_for_bucket = get_variables_for_bucket(i, bucket); in get_grad_buckets() local949 auto variables_for_bucket = get_variables_for_bucket(next_bucket_, bucket); in all_reduce_bucket() local999 std::vector<at::Tensor> variables_for_bucket; in get_variables_for_bucket() local