Searched defs:number_split (Results 1 – 4 of 4) sorted by relevance
29 void SplitWithOverlap::Init(int64_t number_split, const std::vector<int64_t> &ratio, in Init()54 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
25 PrimitivePtr MindIR_SplitWithOverlap_CreatePrimitive(int64_t split_dim, int64_t number_split, const… in MindIR_SplitWithOverlap_CreatePrimitive()80 void MindIR_SplitWithOverlap_SetNumberSplit(PrimitivePtr *primitive, int64_t number_split) { in MindIR_SplitWithOverlap_SetNumberSplit()