class Tensorflow::Error::InvalidArgumentError