Searched defs:split_index (Results 1 – 4 of 4) sorted by relevance
312 def _inner_compute(split_index): argument334 def _inner_compute(split_index, block_idx): argument
467 size_t const split_index = data_length / 2u; in augmented_crc_test() local
442 size_t split_index = parameters.get_min_elements(); in apply() local
2411 …_t> GenerateBucketSizeList(const KernelGraphPtr &graph, const std::vector<uint32_t> &split_index) { in GenerateBucketSizeList()2433 void CheckSplitIndexValid(const vector<uint32_t> &split_index) { in CheckSplitIndexValid()2443 void PreProcessOnSplitIndex(const KernelGraphPtr &graph, vector<uint32_t> *split_index) { in PreProcessOnSplitIndex()2484 auto split_index = GetAllReduceSplitIndex(); in InitAllBucket() local