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

T2 in (tensor(int64)) : Constrain index tensor to int64