ValueDepend
函数功能
标识该输入是否为“数据依赖输入”,数据依赖输入是指在Tiling/InferShape等函数实现时依赖该输入的具体数据。该输入数据为host侧数据,开发者在Tiling函数/InferShape函数中可以通过TilingContext类的GetInputTensor/InferShapeContext类的GetInputTensor获取这个输入数据。
函数原型
OpParamDef &ValueDepend(Option value_depend);
OpParamDef &ValueDepend(Option value_depend, DependScope scope);
参数说明
参数 |
输入/输出 |
说明 |
---|---|---|
value_depend |
输入 |
value_depend有以下两种取值:
|
scope |
输入 |
scope类型为枚举DependScope,支持的取值为:
|
返回值说明
OpParamDef算子定义,OpParamDef请参考OpParamDef。
约束说明
仅支持对算子输入配置,且仅支持输入的参数数据类型配置为DT_INT64/DT_FLOAT/DT_BOOL,对应生成的输出类型分别为aclIntArray、aclFloatArray和aclBoolArray。
父主题: OpParamDef