check_node_support_on_aicore
产品支持情况
全量芯片支持。
功能说明
对传入的Node做校验,校验其是否支持在AI Core上执行。
函数原型
1 | check_node_support_on_aicore(node: Node) -> Tuple[bool, str] |
参数说明
参数名 |
输入/输出 |
描述 |
|---|---|---|
node |
输入 |
待检查的节点,类型为Node。 |
返回值说明
- (Tuple[bool, str]) 返回(is_supported, unsupported_reason)元组。
- is_supported为True表示是否支持在AI Core上执行,False表示不支持。
- unsupported_reason为不支持的具体原因。
约束说明
- 如果检查操作失败,抛出RuntimeError。
- 在调用此接口之前,请确保节点的shape已被推导过。因为校验过程需要结合shape信息进行判断。
父主题: GetUtils