T in (tensor(bfloat16)tensor(double)tensor(float)tensor(float16)tensor(int32)tensor(int64)tensor(uint32),
tensor(uint64)) : Constrain input and output types to float/int tensors.