调用aclCreateTensorDesc接口创建tensor描述信息后,可通过aclSetTensorDynamicInput接口设置多个动态输入的分组标识,例如某个算子包括动态输入为x、必选输入y,其中动态输入有多个,调用aclSetTensorDynamicInput将dynamicInputName设置为x,AscendCL内部根据dynamicInputName索引到对应的多个动态输入,达到分组的目的。同步接口。
aclError aclSetTensorDynamicInput(aclTensorDesc *desc, const char *dynamicInputName)
参数名 |
输入/输出 |
说明 |
---|---|---|
desc |
输出 |
aclTensorDesc类型的指针。 需提前调用aclCreateTensorDesc接口创建aclTensorDesc类型。 |
dynamicInputName |
输入 |
desc中动态输入分组标识的指针。 |
返回0表示成功,返回其它值表示失败。