调用aclCreateTensorDesc接口创建tensor描述信息后,可通过本接口设置tensor的数据值的范围。同步接口。
使用场景:部分算子input的值就是该算子的输出Shape,在动态Shape场景下,这个Shape的值就有一个取值范围,在执行算子时,需要在input上设置Shape值的范围(例如:[[16,16],[1,128],[20,20],[1,10]]),AscendCL才能正常编译、执行算子。
aclError aclSetTensorValueRange(aclTensorDesc* desc, size_t valueCount, int64_t valueRange[][ACL_TENSOR_VALUE_RANGE_NUM])
参数名 |
输入/输出 |
说明 |
---|---|---|
desc |
输出 |
aclTensorDesc类型的指针。 需提前调用aclCreateTensorDesc接口创建aclTensorDesc类型。 |
valueCount |
输入 |
需设置范围的数据值的个数。 |
valueRange |
输入 |
valueRange为每个数据值的范围,用二维数组表示范围。 #define ACL_TENSOR_VALUE_RANGE_NUM 2 |
返回0表示成功,返回其它值表示失败。