INFER_SHAPE

宏功能

针对指定算子，运行其InferShape函数，推导输出shape。

宏原型

INFER_SHAPE(KERNEL_NAME, op_args...)

参数说明

约束说明

• 如果算子需要INFER_SHAPE，那么此宏需要在ADD_TO_LAUNCHER_LIST_AICORE之前调用。
• 如下接口是上述宏定义会调用到的关联接口。

  OP_INPUT(x...)
  OP_OUTPUT(x...)
  OP_ATTR(x...)
  OP_WORKSPACE(x...)
  OP_OUTSHAPE(x...)
  OP_OPTION(x...)
  OP_EMPTY_ARG
  OP_MODE(x...)

调用示例

1
2

// 调用INFER_SHAPE推导算子的输入shape，其中BatchMatMulV3是算子名，OP_INPUT是算子输入参数，OP_OUTPUT是算子输出参数，OP_ATTR是算子的属性参数
INFER_SHAPE(BatchMatMulV3, OP_INPUT(x1, x2, bias, nullptr), OP_OUTPUT(bmmOut), OP_ATTR(adjX1, adjX2, offsetX, opImplModeEnum));