Searched refs:pre_dims_ (Results 1 – 4 of 4) sorted by relevance
44 para->pre_dims_ = 1; in ReSize()53 para->pre_dims_ *= input->DimensionSize(i); in ReSize()
40 unstack_param_.pre_dims_ = 1; in InitKernel()46 unstack_param_.pre_dims_ *= SizeToInt(input_shape[i]); in InitKernel()
29 int pre_dims_; member
27 for (int i = 0; i < para->pre_dims_; i++) { in Unstack()