Intel® oneAPI Deep Neural Network Developer Guide and Reference
A newer version of this document is available. Customers should click here to go to the newest version.
Visible to Intel only — GUID: GUID-3AEF6BAA-88B6-42BA-B126-334C58F32080
Visible to Intel only — GUID: GUID-3AEF6BAA-88B6-42BA-B126-334C58F32080
Square
General
Square operation performs element-wise square operation with given tensor.
Operation attributes
Square operation does not support any attribute.
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 |
src |
Required |
Outputs
Index |
Argument Name |
Required or Optional |
---|---|---|
0 |
dst |
Required |
Supported data types
Square operation supports the following data type combinations.
Src |
Dst |
---|---|
f32 |
f32 |
f16 |
f16 |
bf16 |
bf16 |