TypeError: Cannot handle this data type: (1, 1, 128),