Searched refs:axis_value_pos (Results 1 – 1 of 1) sorted by relevance
845 uint64_t axis_value_pos = LongToUlong(GetPositiveAxis(axis_value, LongToSize(rank))); in InferImplSplit() local847 …if ((x_shape[axis_value_pos] != Shape::SHP_ANY) && (x_shape[axis_value_pos] % output_num_value != … in InferImplSplit()848 MS_LOG(EXCEPTION) << "x_shape[" << axis_value_pos << "] = " << x_shape[axis_value_pos] in InferImplSplit()853 if (output_shape[axis_value_pos] != Shape::SHP_ANY) { in InferImplSplit()854 output_shape[axis_value_pos] = static_cast<int>(x_shape[axis_value_pos] / output_num_value); in InferImplSplit()857 …output_shape_min[axis_value_pos] = static_cast<int>(x_shape_min[axis_value_pos] / output_num_value… in InferImplSplit()859 …output_shape_max[axis_value_pos] = static_cast<int>(x_shape_max[axis_value_pos] / output_num_value… in InferImplSplit()