FilesLast commitLast update
2 months ago
3 months ago
11 days ago
3 days ago
7 months ago
2 months ago
11 days ago
3 days ago
1 month ago
9 days ago
README

SqrtGrad

English | 简体中文

Product Support Status

Product Supported
Ascend 950PR&950DT products √
Atlas A3 products √
Atlas A2 products √
Atlas 200I/500 A2 inference products ×
Atlas inference products √
Atlas training products √

Function Description

  • Operator function: computes the square root gradient.

  • Computation formula:

z=dy∗0.5/yz = dy * 0.5 / y

Parameter Description

Parameter Name Input/Output/Attribute Description Data Type Data Format
y Input Input parameter for the sqrt_grad computation, which is y in the formula. BFLOAT16, FLOAT16, FLOAT ND
dy Input Input parameter for the sqrt_grad computation, which is dy in the formula. BFLOAT16, FLOAT16, FLOAT ND
z Output Output parameter for the sqrt_grad computation, which is z in the formula. BFLOAT16, FLOAT16, FLOAT ND

Constraint Description

None

Invocation Description

Invocation Method Invocation Sample Description
Graph-mode invocation test_geir_sqrt_grad Invokes the SqrtGrad operator by building a graph through the operator IR.