Searched defs:split_size (Results 1 – 5 of 5) sorted by relevance
49 int split_size = split_sizes[split_which]; in DoSplit() local
77 int split_size = input_shape.at(param->split_dim_) / param->num_split_; in CheckAndInitSplitParam() local
74 int split_size = input_shape_[param.split_dim_] / SizeToInt(output_num_); in LaunchSplit() local
81 void CreateOutputShapeAndTypeId(const CNodePtr &origin_cnode, int64_t split_dim, int64_t split_size… in CreateOutputShapeAndTypeId()
408 bool UpdateRatioWithPadStride(int64_t *ratio, size_t ratio_len, size_t split_size, int split_dim_si… in UpdateRatioWithPadStride()