Visible to Intel only — GUID: GUID-76CD896E-A729-4291-99B5-9FFCDA504CB3
Visible to Intel only — GUID: GUID-76CD896E-A729-4291-99B5-9FFCDA504CB3
LogSoftmaxBackward
General
LogSoftmaxBackward operation computes gradient for LogSoftmax.
Operation attributes
Attribute Name |
Description |
Value Type |
Supported Values |
Required or Optional |
---|---|---|---|---|
Represents the axis of which the LogSoftmax is calculated. Negative value means counting dimensions from the back. |
s64 |
Arbitrary s64 value ( -1 in default) |
Optional |
Execution arguments
The inputs and outputs must be provided according to below index order when constructing an operation.
Inputs
Index |
Argument Name |
Required or Optional |
---|---|---|
0 |
diff_dst |
Required |
1 |
dst |
Required |
Outputs
Index |
Argument Name |
Required or Optional |
---|---|---|
0 |
diff_src |
Required |
Supported data types
LogSoftmaxBackward operation supports the following data type combinations.
Diff_dst |
Dst |
Diff_src |
---|---|---|
f32 |
f32 |
f32 |
bf16 |
bf16 |
bf16 |
f16 |
f16 |
f16 |