tfma.model_util.get_model_spec

Returns model spec with given model name.