Searched refs:kDefaultFirstBucketBytes (Results 1 – 3 of 3) sorted by relevance
29 constexpr int kDefaultFirstBucketBytes = int(1024 * 1024); variable
561 py::arg("first_bucket_bytes_cap") = ::c10d::kDefaultFirstBucketBytes, in c10d_init()3168 module.attr("_DEFAULT_FIRST_BUCKET_BYTES") = ::c10d::kDefaultFirstBucketBytes; in c10d_init()
51 first_bucket_types_cap: int = ..., # kDefaultFirstBucketBytes in reducer.hpp