已合并
ClipByNormNoDivSum 算子原型增加去重宏保护 #4544
songkai111创建于 4 天前
ClipByNormNoDivSum 算子原型增加去重宏保护 #4544
已合并
共 1 个文件变更+3-0
| @@ -30,6 +30,8 @@ namespace ge { | |||
| 30 | *@par Outputs: | 30 | *@par Outputs: |
| 31 | * @li y: An ND Tensor. Same dtype and shape as broadcasted inputs. | 31 | * @li y: An ND Tensor. Same dtype and shape as broadcasted inputs. |
| 32 | */ | 32 | */ |
| 33 | |||
| 34 | |||
| 33 | REG_OP(ClipByNormNoDivSum) | 35 | REG_OP(ClipByNormNoDivSum) |
| 34 | .INPUT(x, TensorType({DT_FLOAT16, DT_FLOAT})) | 36 | .INPUT(x, TensorType({DT_FLOAT16, DT_FLOAT})) |
| 35 | .INPUT(greater_zeros, TensorType({DT_FLOAT16, DT_FLOAT})) | 37 | .INPUT(greater_zeros, TensorType({DT_FLOAT16, DT_FLOAT})) |
| @@ -37,6 +39,7 @@ REG_OP(ClipByNormNoDivSum) | |||
| 37 | .INPUT(maximum_ones, TensorType({DT_FLOAT16, DT_FLOAT})) | 39 | .INPUT(maximum_ones, TensorType({DT_FLOAT16, DT_FLOAT})) |
| 38 | .OUTPUT(y, TensorType({DT_FLOAT16, DT_FLOAT})) | 40 | .OUTPUT(y, TensorType({DT_FLOAT16, DT_FLOAT})) |
| 39 | .OP_END_FACTORY_REG(ClipByNormNoDivSum) | 41 | .OP_END_FACTORY_REG(ClipByNormNoDivSum) |
| 42 | |||
| 40 | 43 | ||
| 41 | } // namespace ge | 44 | } // namespace ge |
| 42 | 45 | ||