T1 in (tensor(bfloat16)tensor(double)tensor(float)tensor(float16)) : Constrain grid types to float tensors.

T2 in (tensor(int64)) : Constrain size’s type to int64 tensors.