diff --git a/operator_contrib/LpNormV2CustomSample/FrameworkLaunch/LpNormV2Custom/op_host/lp_norm_v2_custom.cpp b/operator_contrib/LpNormV2CustomSample/FrameworkLaunch/LpNormV2Custom/op_host/lp_norm_v2_custom.cpp index e7fd7aba046607173dbbdccfca9fc495375f800f..b46a71d570c37c35f83492ac6d9a2a84a9e1718e 100644 --- a/operator_contrib/LpNormV2CustomSample/FrameworkLaunch/LpNormV2Custom/op_host/lp_norm_v2_custom.cpp +++ b/operator_contrib/LpNormV2CustomSample/FrameworkLaunch/LpNormV2Custom/op_host/lp_norm_v2_custom.cpp @@ -66,7 +66,9 @@ static ge::graphStatus TilingFunc(gert::TilingContext* context) // 处理多轴情况,axes为空时,不需要计算 // 仅处理了axes列表中为连续维度的情况,例如[0], [1, 2]之类 // 因为是连续维度,所以定位通过axes列表的值,将shape中的维度分为三部分 - // 前部分(unitCount),中间部分(stepSize),后部分(totalLength) + // unitCount 需要分别处理多少组元素 + // totalLength 每组要处理的元素的总个数 + // stepSize 每组中要处理的元素的相邻下标的距离 auto shape = context->GetInputTensor(0)->GetShape().GetOriginShape(); uint32_t dimNum = shape.GetDimNum(); uint32_t maxdim = 0;