Update the data types and tensor operations for training
1 file changed