T in (tensor(double)tensor(float)tensor(bfloat16)tensor(float16)) : Constrain input and output types to float tensors.