INFER_SHAPE

宏功能

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

宏原型

INFER_SHAPE(KERNEL_NAME, op_args...)

如下接口是上述宏定义会调用到的关联接口。

OP_INPUT(x...)

OP_OUTPUT(x...)

OP_ATTR(x...)

OP_WORKSPACE(x...)

OP_OUTSHAPE(x...)

OP_OPTION(x...)

OP_EMPTY_ARG

OP_MODE(x...)

参数说明

约束说明

如果算子需要INFER_SHAPE，那么此宏需要在ADD_TO_LAUNCHER_LIST_AICORE之前调用。

调用示例

// 调用INFER_SHAPE推导batchmatmul算子的输入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));