Shape fix for LFADS tf.case statements
Shape fix for LFADS tf.case statements

sshleifer/object_detection_kitti,derekjchow/models,xcyan/models,xcyan/models,alexgorban/models,tombstone/models,jch1/models,zbigniewwojna/text-rcnn,jiaphuan/models,cshallue/models,cshallue/models,alexgorban/models,xcyan/models,derekjchow/models,jiaphuan/models,sshleifer/object_detection_kitti,jch1/models,jch1/models,xcyan/models,jch1/models,jch1/models,tombstone/models,alexgorban/models,alexgorban/models,sshleifer/object_detection_kitti,tombstone/models,jch1/models,sshleifer/object_detection_kitti,xcyan/models,cshallue/models,cshallue/models,derekjchow/models,alexgorban/models,cshallue/models,tombstone/models,derekjchow/models,jiaphuan/models,jiaphuan/models,sshleifer/object_detection_kitti,jiaphuan/models,tombstone/models,derekjchow/models,jiaphuan/models,xcyan/models,sshleifer/object_detection_kitti,tombstone/models