Searched refs:gather_param (Results 1 – 1 of 1) sorted by relevance
32 auto *gather_param = reinterpret_cast<GatherParameter *>(malloc(sizeof(GatherParameter))); in PopulateGatherParameter() local33 if (gather_param == nullptr) { in PopulateGatherParameter()37 memset(gather_param, 0, sizeof(GatherParameter)); in PopulateGatherParameter()38 gather_param->op_parameter_.type_ = schema::PrimitiveType_Gather; in PopulateGatherParameter()41 free(gather_param); in PopulateGatherParameter()44 gather_param->axis_ = gather_prim->axis(); in PopulateGatherParameter()45 return reinterpret_cast<OpParameter *>(gather_param); in PopulateGatherParameter()