Versioned name: Square-1
Short description: Square performs element-wise square operation with given tensor.
input- input tensor. Required.
output- the output tensor of element-wise square operation. Required.
T: f32, f16, bf16.
Note: Inputs and outputs have the same data type denoted by T. For example, if input is f32 tensor, then all other tensors have f32 data type.