where
功能介绍 |
根据条件返回x或y中的元素。 |
||
---|---|---|---|
函数原型 |
|
||
用法 |
用法同TF原生的where算子,condition为bool类型,x和y为特定类型入参(tf.int64)会执行SVE优化后的算子;其他类型入参则执行原生的where算子。 具体安装方法可参考Rec SDK TensorFlow源码中的“cust_op/tf_cpu_op/README.md”。 |
||
约束说明 |
|
父主题: Host算子
功能介绍 |
根据条件返回x或y中的元素。 |
||
---|---|---|---|
函数原型 |
|
||
用法 |
用法同TF原生的where算子,condition为bool类型,x和y为特定类型入参(tf.int64)会执行SVE优化后的算子;其他类型入参则执行原生的where算子。 具体安装方法可参考Rec SDK TensorFlow源码中的“cust_op/tf_cpu_op/README.md”。 |
||
约束说明 |
|