%1 = "tfl.reshape"(%0, %cst) : (tensor<2x200x200x200xf32>, tensor<5xi64>) -> tensor<2x1x200x200x200xf32>
    func.return %1 : tensor<2x1x200x200x200xf32>
  }