Searched defs:split_dim_size (Results 1 – 4 of 4) sorted by relevance
50 int split_dim_size = input_shape[split_dim]; in SplitWithOverlapInferShape() local
39 auto split_dim_size = shape[param_->split_dim_]; in CalculateSplitedShapes() local
38 void SplitConstantData(char *in_data, char **out_data, int64_t num_split, int64_t split_dim_size, i… in SplitConstantData()64 int64_t split_dim_size = constant_shape[split_dim]; in CreateSplitConstantTensors() local
408 …atioWithPadStride(int64_t *ratio, size_t ratio_len, size_t split_size, int split_dim_size, int pad, in UpdateRatioWithPadStride()