全量芯片支持。
#include <graph/operator_reg.h>
提供公共函数宏封装,供算子开发者开发InferShape函数。该函数基于2个输入的shape,设置输出的shape。该宏只是设置shape,未设置dtype。
如果2个输入的shape一致,会按输入的shape设置输出shape。
如果2个输入的shape不一致,会按照Broadcast的策略,取2个输入shape的并集。
比如输入shape分别为(1,2,3,4)和(3,1,3,4),则该宏会设置算子的输出shape为(3,2,3,4)。
[object Object]
该函数会自动调用如下函数:
[object Object]
执行成功或失败。
无
[object Object]