ELMTWISE_INFER_SHAPEANDTYPE
函数功能
提供公共函数宏封装，供算子开发者开发InferShape函数。该函数基于输入的shape和dtype，设置输出的shape和dtype。
例如，输入shape为（1,2,3,4），dtype为float，则该宏会设置算子的输出shape为（1,2,3,4），输出dtype为float。
函数原型
ELMTWISE_INFER_SHAPEANDTYPE(in_name, out_name)
约束说明
无
参数说明
| 参数名 | 输入/输出 | 描述 |
|---|---|---|
| in_name | 输入 | 算子输入。 |
| out_name | 输入 | 算子输出。 |
返回值
执行成功或失败。
调用示例
COMMON_INFER_FUNC_REG(DiagD, ELMTWISE_INFER_SHAPEANDTYPE("assist", "y"));