开发者
下载
[object Object][object Object]

全量芯片支持。

[object Object]

#include <graph/operator_reg.h>

[object Object]

提供公共函数宏封装,供算子开发者开发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]
[object Object][object Object]undefined
[object Object]

执行成功或失败。

[object Object]

[object Object]
[object Object]