Searched refs:unsqueeze_param (Results 1 – 1 of 1) sorted by relevance
32 …auto *unsqueeze_param = reinterpret_cast<UnSqueezeParameter *>(malloc(sizeof(UnSqueezeParameter))); in PopulateUnsqueezeParameter() local33 if (unsqueeze_param == nullptr) { in PopulateUnsqueezeParameter()37 memset(unsqueeze_param, 0, sizeof(UnSqueezeParameter)); in PopulateUnsqueezeParameter()38 unsqueeze_param->op_parameter_.type_ = schema::PrimitiveType_Unsqueeze; in PopulateUnsqueezeParameter()42 free(unsqueeze_param); in PopulateUnsqueezeParameter()47 free(unsqueeze_param); in PopulateUnsqueezeParameter()50 unsqueeze_param->num_dim_ = static_cast<int>(flat_axis->size()); in PopulateUnsqueezeParameter()53 unsqueeze_param->dims_[i++] = *iter; in PopulateUnsqueezeParameter()55 return reinterpret_cast<OpParameter *>(unsqueeze_param); in PopulateUnsqueezeParameter()