Searched defs:number_split (Results 1 – 3 of 3) sorted by relevance
21 void SplitWithOverlap::Init(int64_t number_split, const std::vector<int64_t> &ratio, in Init()44 void SplitWithOverlap::set_number_split(int64_t number_split) { in set_number_split()
38 auto number_split = attr_value.i(); in Parse() local
35 int number_split = param->num_split_; in SplitWithOverlapInferShape() local